题目描述
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.
解题思路
按以下步骤进行:
- root == null,则返回false;
- root是叶子节点,如果root.val == sum,返回true,否则返回false;
- 如果root不是叶子节点 && root.left!=null,如果hasPathSum(root.left,sum - root.val) == true,则返回true,否则如果root.right!=null,返回hasPathSum(root.right,sum - root.val);
- 如果root不是叶子节点 && root.left==null,返回hasPathSum(root.right,sum - root.val);
需要注意的是,虽然示例中给出的全是正整数,但是测试用例时会给出负整数。
代码
public static boolean hasPathSum(TreeNode root, int sum) {
if(root==null){
return false;
}
if(root.left==null && root.right==null){
if(root.val==sum){
return true;
} else {
return false;
}
}
if(root.left!=null){
if(hasPathSum(root.left,sum - root.val)){
return true;
} else {
if(root.right!=null){
return hasPathSum(root.right,sum - root.val);
}
}
} else {
return hasPathSum(root.right,sum - root.val);
}
return false;
}