-
Notifications
You must be signed in to change notification settings - Fork 57
/
Copy pathSolution.java
56 lines (55 loc) · 1.48 KB
/
Solution.java
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
48
49
50
51
52
53
54
55
56
/**
* You are given a binary tree in which each node contains an integer value.
*
* Find the number of paths that sum to a given value.
*
* The path does not need to start or end at the root or a leaf, but it must go downwards (traveling only from parent nodes to child nodes).
*
* The tree has no more than 1,000 nodes and the values are in the range -1,000,000 to 1,000,000.
*
* Example:
*
* root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8
*
* 10
* / \
* 5 -3
* / \ \
* 3 2 11
* / \ \
* 3 -2 1
*
* Return 3. The paths that sum to 8 are:
*
* 1. 5 -> 3
* 2. 5 -> 2 -> 1
* 3. -3 -> 11
*/
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int pathSum(TreeNode root, int sum) {
return pathSumCheck(root, sum, false);
}
int pathSumCheck(TreeNode root, int sum, boolean includeOnly) {
if (root == null) {
return 0;
}
int self = (root.val == sum) ? 1 : 0;
int left2 = pathSumCheck(root.left, sum - root.val, true);
int right2 = pathSumCheck(root.right, sum - root.val, true);
if (includeOnly) {
return self + left2 + right2;
}
int left1 = pathSumCheck(root.left, sum, false);
int right1 = pathSumCheck(root.right, sum, false);
return self + left2 + right2 + left1 + right1;
}
}