思路
- 我的初始思路,一个int add做当前的和,再用dfs遍历所有节点,满了就减,减了再加,直到add==sum并且正好到叶子节点
- 优化:子节点给个累计值,是父节点到root的总和,只需要比较大小即可http://bowang.name/leetcode-150/
- *
Attention
- 当tree为null和sum为0时候
- 当节点有负数的时候
一个recursive常出现的bug
public boolean hasPathSum(TreeNode root, int sum) {
if (root == null) return false;
return leafSum(root, 0, sum);
}
private static boolean leafSum (TreeNode root, int parSum, int sum) {
int currSum = parSum + root.val;
if (root.left == null && root.right == null && currSum == sum) {
return true;
}
if (root.left != null) {
return leafSum(root.left, currSum, sum);
}
if (root.right != null) {
return leafSum(root.right, currSum, sum);
}
return false;
}
对于这个代码,总是先看左边,再看右边。一旦左边为false就终止了。
private static boolean leafSum (TreeNode root, int parSum, int sum) {
int currSum = parSum + root.val;
if (root.left == null && root.right == null && currSum == sum) {
return true;
}
if (root.left != null && leafSum(root.left, currSum, sum)) {
return true;
}else if (root.right != null && leafSum(root.right, currSum, sum)) {
return true;
} else {
return false;
}
}
或者
private static boolean leftSum2(TreeNode root, int parSum, int sum) {
int currSum = parSum + root.val;
if (root.left == null && root.right == null && currSum == sum) {
return true;
}
return root.left != null && leafSum(root.left, currSum, sum)
|| root.right != null && leafSum(root.right, currSum, sum);
}
对于第二种优化还有更好的方法,就是设置ans1和ans2作为左右子树的return值, default值都设为false,最后返回ans1 || ans2
就行了.