111. 二叉树的最小深度

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

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

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

示例 1：

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

示例 2：

1
2
输入：root = [2,null,3,null,4,null,5,null,6]
输出：5

提示：

  • 树中节点数的范围在 [0, 105] 内
  • -1000 <= Node.val <= 1000

题解：

递归：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
public int minDepth(TreeNode root) {
    //递归出口
    if(root == null){
        return 0;
    }
    //左右子树有为空的情况，返回非空子树的最小深度+1
    if(root.left == null && root.right != null){
        return 1 + minDepth(root.right);
    }

    if(root.left != null && root.right == null){
        return 1 + minDepth(root.left);
    }
    //左右子树都不为空时
    return 1 + Math.min(minDepth(root.left),minDepth(root.right));

}

迭代：

使用层序遍历找到第一个叶子结点，就得到了最小深度

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
public int minDepth(TreeNode root) {

    if (root == null) {
        return 0;
    }

    Queue<TreeNode> queue = new ArrayDeque<>();

    queue.offer(root);
    //记录深度，初始为0
    int depth = 0;

    while (!queue.isEmpty()) {
        depth++;
        int length = queue.size();

        for (int i = 0; i < length; i++) {

            TreeNode node = queue.poll();
            //找到了第一个叶子结点，即最小深度，直接返回depth
            if (node.left == null && node.right == null) {
                return depth;
            }

            if (node.left != null) {
                queue.offer(node.left);
            }
            if (node.right != null) {
                queue.offer(node.right);
            }
        }
    }
    return depth;

}