104. 二叉树的最大深度

Posted on | In
Words count in article: 336 | Reading time ≈ 1

题目

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

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

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

示例1：

给定二叉树 [3,9,20,null,null,15,7]，

1
2
3
4
5
  3
 / \
9  20
  /  \
 15   7

返回它的最大深度 3 。

解法

解法一：递归求解

分别求出左右子树的最大深度，最后加一就行。

Java

1
2
3
4
5
6
7
8
9
public int maxDepth(TreeNode root) {
    if (null == root) {
        return 0;
    }
		
    int dl = maxDepth(root.left) + 1;
    int dr = maxDepth(root.right) + 1;
    return dl > dr ? dl : dr;
    }

解法二：借助栈

我们还可以在栈的帮助下将上面的递归转换为迭代。

使用 DFS 策略访问每个结点，同时在每次访问时更新最大深度。

所以我们从包含根结点且相应深度为 1 的栈开始。然后我们继续迭代：将当前结点弹出栈并推入子结点。每一步都会更新深度.

Java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
public int maxDepth(TreeNode root) {
    Queue<Pair<TreeNode, Integer>> stack = new LinkedList<>();
    if (root != null) {
      stack.add(new Pair(root, 1));
    }

    int depth = 0;
    while (!stack.isEmpty()) {
      Pair<TreeNode, Integer> current = stack.poll();
      root = current.getKey();
      int current_depth = current.getValue();
      if (root != null) {
        depth = Math.max(depth, current_depth);
        stack.add(new Pair(root.left, current_depth + 1));
        stack.add(new Pair(root.right, current_depth + 1));
      }
    }
    return depth;
  }
0%