maxDepth

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

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

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

示例:
给定二叉树 [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;
        if(root.left != null && root.right != null)
            return Math.max(maxDepth(root.left), maxDepth(root.right)) + 1;
        return maxDepth(root.left) + maxDepth(root.right) + 1;
    }
}

// 迭代
class Solution {
    public int maxDepth(TreeNode root) {
        Queue<Pair<TreeNode, Integer>> queue = new LinkedList<>();
        int max = 0;
        if(root != null)
            queue.add(new Pair<>(root, 1));
        while(!queue.isEmpty()){
            Pair<TreeNode, Integer> pair = queue.poll();
            TreeNode root1 = pair.getKey();
            int cnt = pair.getValue();
            if(root1 != null){
                max = Math.max(max, cnt);
                queue.add(new Pair<>(root1.left, cnt+1));
                queue.add(new Pair<>(root1.right, cnt+1));
            }
        }
        return max;
    }
}
posted @ 2020-06-05 09:33  athony  阅读(243)  评论(0编辑  收藏  举报