102 字 少于 1 分钟阅读

给定一个二叉树,找出其最大深度。

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

说明: 叶子节点是指没有子节点的节点。

示例: 给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7

返回它的最大深度 3 。

解法

遍历的时候带上当前节点的深度即可。

class Solution {
    public int maxDepth(TreeNode root) {
        if (root == null){
            return 0;
        }
        return traversal(root, 0);
    }

    public int traversal(TreeNode node, int depth) {
        if (node == null){
            return depth;
        }
        depth++;
        int leftDepth = traversal(node.left, depth);
        int rightDepth = traversal(node.right, depth);
        return Math.max(leftDepth, rightDepth);
    }
}

优化

class Solution {
    public int maxDepth(TreeNode root) {
        if (root == null){
            return 0;
        }
        int leftDepth = maxDepth(root.left);
        int rightDepth = maxDepth(root.right);
        //加1的意思是当前节点加上左右子树的最大深度为整体最大深度
        return Math.max(leftDepth, rightDepth) + 1;
    }
}