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.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47 |
/** * Definition for binary tree * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */public
class Solution { /** * This problem can be solved either by dfs or bfs.<br> * @param root * @param sum * @return */ public
boolean hasPathSum(TreeNode root, int
sum) { boolean
exists = false; if(root != null){ Map<TreeNode, Integer> sumAtNode = new
HashMap<TreeNode, Integer>(); sumAtNode.put(root, root.val); Deque<TreeNode> nodes = new
LinkedList<TreeNode>(); nodes.add(root); while(nodes.peek()!= null){ TreeNode aNode = nodes.poll(); if(aNode.left == null
&& aNode.right == null){ if(sumAtNode.get(aNode) == sum){ exists = true; break; } } else{ int
pareValue = sumAtNode.get(aNode); if(aNode.left != null){ sumAtNode.put(aNode.left, aNode.left.val + pareValue); nodes.add(aNode.left); } if(aNode.right != null){ sumAtNode.put(aNode.right, aNode.right.val + pareValue); nodes.add(aNode.right); } } } } return
exists; }} |
leetcode--Path Sum,布布扣,bubuko.com
原文地址:http://www.cnblogs.com/averillzheng/p/3773952.html