算法通关村——轻松搞定求二叉树深度问题

LeetCode 104 二叉树的最大深度

给定一个二叉树 root ，返回其最大深度。

二叉树的 最大深度 是指从根节点到最远叶子节点的最长路径上的节点数。

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：3

示例 2：

输入：root = [1,null,2]
输出：2 

 首先是递归做法。

在二叉树中，一个节点的深度就是，这个节点的左右子树的最大深度 + 1。

代码：

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

而层序遍历的做法，就是在遍历过程中，计算二叉树的层数，就是二叉树的深度。

代码：

class Solution {
    public int maxDepth(TreeNode root) {
        Deque<TreeNode> queue = new LinkedList<TreeNode>();
        int ans = 0;
        if(root == null)
            return 0;
        queue.add(root);
        while(!queue.isEmpty()){
            int size = queue.size();
            for(int i = 0;i < size;i++){
                TreeNode node = queue.remove();
                if(node.left != null)
                    queue.add(node.left);
                if(node.right != null)
                    queue.add(node.right);
            }
            ans++;
        }
        return ans;
    }
}