题目

类型：DFS

难度：简单

二叉树的最大深度 - 图1

解题思路

1、深度优先搜索先递归计算出其左子树和右子树的最大深度，然后在 O(1) 时间内计算出当前二叉树的最大深度。递归在访问到空节点时退出。

2、广度优先搜索的队列里存放的是「当前层的所有节点」。每次拓展下一层的时候，需要将队列里的所有节点都拿出来进行拓展，这样能保证每次拓展完的时候队列里存放的是当前层的所有节点。用一个变量ans来维护拓展的次数，该二叉树的最大深度即为ans。

代码

深度优先

  1. class Solution {
  2.     public int maxDepth(TreeNode root) {
  3.         if (root == null) {
  4.             return 0;
  5.         } else {
  6.             int leftHeight = maxDepth(root.left);
  7.             int rightHeight = maxDepth(root.right);
  8.             return Math.max(leftHeight, rightHeight) + 1;
  9.         }
  10.     }
  11. }

广度优先

class Solution {
    public int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(root);
        int ans = 0;
        while (!queue.isEmpty()) {
            int size = queue.size();
            while (size > 0) {
                TreeNode node = queue.poll();
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
                size--;
            }
            ans++;
        }
        return ans;
    }
}