Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum.
For example:Given the below binary tree and
sum = 22
,5 / \ 4 8 / / \ 11 13 4 / \ \ 7 2 1
return true, as there exist a root-to-leaf path
5->4->11->2
which sum is 22.
之前考虑的一直都是,怎么从根到叶子的结点权值挨个相加,然后与sum值比较。这样就得考虑叶子个数的数的处理,如何存储或者输出?用递归的话该如何实现?怎么分清楚左右子树。头大。
换个思路,将sum自根节点起逐步减结点权值,最后与叶子的权值比较,这样就好处理多了!!
核心代码就一句,return hasPathSum(root.left, sum - root.val) || hasPathSum(root.right, sum - root.val);
上代码。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
if(root == null)
return false;
if(root.val == sum && root.left == null && root.right == null)
return true;
return hasPathSum(root.left, sum - root.val) || hasPathSum(root.right, sum - root.val);
}
}
没有评论:
发表评论