原题:#111_二叉树最小深度
- 关键思想:
- 叶子节点:无孩子节点
- 当只有root时,深度为1;
- 只存在一个孩子节点时,返回孩子节点深度
- 存在两孩子节点时,返回最小的深度
- 递归:
- 结束条件:根节点为空时结束
- 递归操作:分别查找左右子树的深度
- 返回值:
- 当只有root时,深度为1;
- 只存在一个孩子节点时,返回孩子节点深度
- 存在两孩子节点时,返回最小的深度
class Solution {
public int minDepth(TreeNode root) {
if(root == null) return 0;
//这道题递归条件里分为三种情况
//1.左孩子和有孩子都为空的情况,说明到达了叶子节点,直接返回1即可
if(root.left == null && root.right == null) return 1;
//2.如果左孩子和由孩子其中一个为空,那么需要返回比较大的那个孩子的深度
int m1 = minDepth(root.left);
int m2 = minDepth(root.right);
//这里其中一个节点为空,说明m1和m2有一个必然为0,所以可以返回m1 + m2 + 1;
if(root.left == null || root.right == null) return m1 + m2 + 1;
//3.最后一种情况,也就是左右孩子都不为空,返回最小深度+1即可
return Math.min(m1,m2) + 1;
}
}
- 精简版:
class Solution {
public int minDepth(TreeNode root) {
if(root == null) return 0;
int m1 = minDepth(root.left);
int m2 = minDepth(root.right);
//1.如果左孩子和右孩子有为空的情况,直接返回m1+m2+1
//2.如果都不为空,返回较小深度+1
return root.left == null || root.right == null ? m1 + m2 + 1 : Math.min(m1,m2) + 1;
}
}