[二叉树专题]求二叉树的高度(深度)

/* 递归解法 O(n)
1。若是二叉树为null,则二叉树的深度为0;
2. 若是二叉树不为null,则二叉树的深度=max(左子树的深度,右子树的深度)+1;
public static int getDepthRec(TreeNode root){
    if(root == null){
        return 0;
    }
    
    int leftDepth = getDepthRec(root.left);
    int rightDepth = getDepthRec(root.right);
    return Math.max(leftDepth,rightDepth)+1;
}
/* 迭代解法O(n)
基本思想同LevelOrderTravelsal,仍是用一个Queue。
*/
public static int getDepth(TreeNode root){
    if(root == null){
        return 0;
    }
    
    int depth = 0;//深度
    int currentLevelNodes = 1;//当前level,node的数量
    int nextLevelNodes = 0;//下一层,node的数量
    
    Queue<TreeNode> queue = new LinkedList<TreeNode>();
    queue.add(root);
    while(!queue.isEmpty()){
        TreeNode cur = queue.remove();//从队头的位置移除
        currentLevelNodes--;//减小当前level node数量
        if(cur.left != null)(    //若是有左孩子,则加到队尾
            queue.add(cur.left);
            nextLevelNodes++;//并增长下一层Level Node 的数量
        )
        if(cur.right != null){//若是有右孩子,加到队尾
            queue.add(cur.right);
            nextLevelNodes++;
        }
        
        if(currentLevelNodes == 0){//已经遍历完当前层的全部节点
            depth ++; //增长高度
            currentLevelNodes = nextLevelNodes;//初始化下一层
            nextLevelNodes = 0;
        }
    }
    return depth;
}
相关文章
相关标签/搜索