考虑类似于最大深度的求法,即使用递归。
- 到达叶子节点,即左右节点都为null,返回 0(较小的深度,但两边都是0) + 1;
- 节点的左右子树都不为null,返回左右子树较小的深度 + 1;
- 节点的左右子树只有一方为null时,求法有所不同,要返回的应该是左右子树深度较大的 + 1,而不是较小的。(因为当节点的左右子树只有一方为null时,返回较小的左右子树深度 + 1,会是0 + 1,但当前节点并非叶子节点,不该 从此处开始 计算最小深度,直接返回0 + 1,会导致上一层的以及最终的深度求解 错误。)
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* public TreeNode(int val) {
* this.val = val;
* }
* }
*/
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @return int整型
*/
public int run (TreeNode root) {
if (root == null) {
return 0;
}
/* 1. 广度优先
// 最小深度,考虑广度优先,找到第一个叶子节点出现的层数
int result = 0;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
int size = queue.size();
result++;
while (size-- != 0) {
TreeNode node = queue.poll();
if (node.left == null && node.right == null) {
return result;
}
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
}
}
return result;
*/
// 2. 递归
return minDepth(root);
}
public int minDepth(TreeNode root) {
if (root == null) {
return 0;
}
// 如果左右都为null,即到达叶子节点,返回1,即较小的(都是0) + 1
if (root.left == null && root.right == null) {
return 1;
}
// 如果左右子树只有一个为null,就要返回二者之中深度较大的 + 1
if (root.left == null || root.right == null) {
return Math.max(minDepth(root.left), minDepth(root.right)) + 1;
}
// 如果都不为null,就返回较小的 + 1
return Math.min(minDepth(root.left), minDepth(root.right)) + 1;
}
}

京公网安备 11010502036488号