题目
求二叉树的最小深度:从根节点到叶子节点的路径上
分析
DFS,当判断到达叶子节点时记录depth,并进行最小比较。
这么做了之后发现效率较低,只击败10%的提交。
思考之后发现可以进行剪枝。当当前depth大于等于min时就不需要再遍历了,因为肯定找不出更短的了。
关键:DFS容易想到,关键是剪枝!
代码
class Solution {
public int minDepth(TreeNode root) {
int[] min = {0};
dfs(root, min, 0);
return min[0];
}
private void dfs(TreeNode root,int[] min,int depth) {
if(root == null || (min[0] != 0 && depth >= min[0])) {
return;
}
if(root.left == null && root.right == null) {
if(min[0] == 0) {
min[0] = depth + 1;
} else {
min[0] = Math.min(min[0], depth + 1);
}
return;
}
if(root.left != null) {
dfs(root.left, min, depth + 1);
}
if(root.right != null) {
dfs(root.right, min, depth + 1);
}
}
}