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
Solution: recursion5->4->11->2
which sum is 22.
时间复杂度是O(n),空间复杂度是O(logn)。
/** * Definition for binary tree * 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; int diff = sum - root.val; if(root.left == null && root.right == null) return (diff == 0); else return (hasPathSum(root.left, diff)) || (hasPathSum(root.right, diff)); } }
No comments:
Post a Comment