代码随想录day16104. 二叉树的最大深度

代码随想录day16|104. 二叉树的最大深度

104. 二叉树的最大深度

//迭代法
class Solution {
    public int maxDepth(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        int count = 0;
        if(root == null)return 0;
            queue.offer(root);
        
        while(!queue.isEmpty()){
            int size = queue.size();
            for(int i = 0 ; i < size ; i++){
                TreeNode node = queue.poll();
                if(node.left!=null)queue.offer(node.left);
                if(node.right != null)queue.offer(node.right);
            }
            count++;
        }
        return count;
    }
}
//递归法

class Solution {
    public int maxDepth(TreeNode root) {
        if(root == null)return 0;
        return Math.max(maxDepth(root.right) , maxDepth(root.left))+1;
    }
}

111. 二叉树的最小深度

//迭代法
class Solution {
    public int minDepth(TreeNode root) {
    Queue<TreeNode> queue = new LinkedList<>();
        int count = 0;
        if(root == null)return 0;
            queue.offer(root);
        
        while(!queue.isEmpty()){
            int size = queue.size();
            count++;
            for(int i = 0 ; i < size ; i++){
                TreeNode node = queue.poll();
                if(node.left==null && node.right == null)return count;
                if(node.left!=null)queue.offer(node.left);
                if(node.right != null)queue.offer(node.right);
            }
            
        }
        return count;
    }
}
//递归
class Solution {
    /**
     * 递归法，相比求MaxDepth要复杂点
     * 因为最小深度是从根节点到最近**叶子节点**的最短路径上的节点数量
     */
    public int minDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int leftDepth = minDepth(root.left);
        int rightDepth = minDepth(root.right);
        if (root.left == null) {
            return rightDepth + 1;
        }
        if (root.right == null) {
            return leftDepth + 1;
        }
        // 左右结点都不为null
        return Math.min(leftDepth, rightDepth) + 1;
    }
}

222. 完全二叉树的节点个数

class Solution {
     public int countNodes(TreeNode root) {
        if(root == null) {
            return 0;
        }
        int leftDepth = getDepth(root.left);
        int rightDepth = getDepth(root.right);
        if (leftDepth == rightDepth) {// 左子树是满二叉树
            // 2^leftDepth其实是 （2^leftDepth - 1） + 1 ，左子树 + 根结点
            return (1 << leftDepth) + countNodes(root.right);
        } else {// 右子树是满二叉树
            return (1 << rightDepth) + countNodes(root.left);
        }
    }

    private int getDepth(TreeNode root) {
        int depth = 0;
        while (root != null) {
            root = root.left;
            depth++;
        }
        return depth;
    }

}