Leetcode111:给定一个二叉树,找出最小深度。
最小深度的定义是:从根节点到叶子结点的最短路径上的节点个数。
思路1:递归方式解决最小深度
如图,以值为5的节点为例,如果要找递推公式,我们想到的最简单最初始的递推公式就是——
最小深度 == Math.min(左子树最小深度,右子树最小深度) + 1
因为最小深度对应一条路径,既然左右两棵子树都有一条最短路径,那么根节点的最短路径肯定就是子树最短路径再加上1。
递推终止条件怎么写呢?我们知道总有一个时候会递归到null节点,null节点没有左右子树了,不能递归了,应该终止递归,并返回最小深度0.
if(root == null) return 0;
这是最初始的、条件反射的思考。一个大致的递推框架建成后,再考虑特殊情况。为什么说有特殊情况?因为我们是依照上面的图思考的,在上面的图中,节点5的左右子树都不为null;为了使递推公式更严谨,我们还需要考虑左右子树为null的情况,如下图,如果左右子树有一个为null,那还能使用Math,min吗?
显然不能,如果对节点1使用Math.min,它的左子树null又返回了结果0,那这就不对了,因为一条最短路径的终点并不是null,也不是节点1,而应该是叶子结点。所以如果出现了有一边子树为Null而另一边不为Null的情况,就要重新写递推式子,这时最小深度应该是:如果左子树为空,最小深度为右子树最小深度+1;如果右子树为null,那么最小深度为左子树最小深度+1。
我们已经考虑了左子树右子树有一边是null的情况,也考虑了左右子树都不为Null的情况,那都为Null呢?都为Null不就是叶子节点吗,叶子结点最小深度为1。只是这个没有必要再单独拉出来写成一个if判断条件,因为用Math.min(左子树最小深度,右子树最小深度)+1来算,叶子结点深度算出来就是1。所以包含在了公式中,不需额外讨论。
综上所述,代码应该这么写,注意递归终止条件写在前面,递推公式写在后面:
public static int minDepth(TreeNode root){ if (root == null){ return 0; } if (root.left == null && root.right != null){ return minDepth(root.right) + 1; } if (root.left != null && root.right == null){ return minDepth(root.left) + 1; } return Math.min(minDepth(root.left),minDepth(root.right)) + 1; //如果根节点不为Null,左右子树也不是只有一边为null,就是正常情况,返回左右子树最小深度中的最小值加1。 //这里没有单独列出叶子结点的判断,是因为如果是叶子结点就返回1,跟用叶子结点的两个null子树做递归是一样的。 }
还有没有别的方法呢?是有的,你想,最小深度对应路径是根节点和叶子结点路径中,最短的一条。实际上不就是找隔根节点层数最少的叶子结点吗?也就是寻找最上层的叶子结点。那么使用层序遍历的方式一层一层地遍历节点,找到第一个叶子结点就返回当前层数,这个层数就是最小深度,不信可以画图自己看看。
代码如下:
public static int minDepth2(TreeNode root){
if (root == null){
return 0;
}
LinkedList<TreeNode> queue = new LinkedList<>();
queue.add(root);
int res = 0;
//res代表结果,是第多少层。
//这里先使用层序遍历,对于每个出队的元素都判断是否为叶子结点,一旦是就返回当前层数。
while (!queue.isEmpty()){
int size = queue.size();
res++;//res为当前层数,while循环每进行一次都相当于开始遍历下一层,因此每次循环res++
for (int i = 0; i < size; i++) {
TreeNode node = queue.remove();
//判断是否为叶子结点,是就直接返回层数
if (node.left == null && node.right == null){
return res;
}
if (node.left != null){
queue.add(node.left);
}
if (node.right != null){
queue.add(node.right);
}
}
}
return 0;
}