给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值 的路径。
一个有效的路径,指的是从根节点到叶节点的路径。
样例
给定一个二叉树,和 目标值 = 5:
1
2
3
4
5
|
1
/ \
2 4
/ \
2 3
|
返回:
1
2
3
4
|
[
[ 1 , 2 , 2 ],
[ 1 , 4 ]
]
|
代码如下:
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
/**
* Definition of TreeNode:
* public class TreeNode {
* public int val;
* public TreeNode left, right;
* public TreeNode(int val) {
* this.val = val;
* this.left = this.right = null;
* }
* }
*/
public class Solution {
/**
* @param root the root of binary tree
* @param target an integer
* @return all valid paths
*/
public List<List<Integer>> binaryTreePathSum(TreeNode root, int target) {
// Write your code here
return dfs(root, new ArrayList<Integer>(), 0 , new ArrayList<List<Integer>>(),target);
}
public List<List<Integer>> dfs(TreeNode root,List<Integer> node, int sum, List<List<Integer>> paths, int target)
{
if (root== null )
{
return new ArrayList<List<Integer>>();
}
List<List<Integer>> path= new ArrayList<List<Integer>>();
if (root.left!= null )
{
List<Integer> nodes= new ArrayList<Integer>();
if (node!= null )
{
nodes.addAll(node);
}
nodes.add(root.val);
List<List<Integer>> temp=dfs(root.left,nodes,sum+root.val,paths,target);
if (temp!= null )
{
path.addAll(temp);
}
}
if (root.right!= null )
{
List<Integer> nodes= new ArrayList<Integer>();
if (node!= null )
{
nodes.addAll(node);
}
nodes.add(root.val);
List<List<Integer>> temp=dfs(root.right,nodes,sum+root.val,paths,target);
if (temp!= null )
{
path.addAll(temp);
}
}
if (root.left== null &&root.right== null )
{
List<Integer> nodes= new ArrayList<Integer>();
if (node!= null )
{
nodes.addAll(node);
}
nodes.add(root.val);
if (sum+root.val==target)
{
path.add(nodes);
} else {
path= new ArrayList<List<Integer>>();
}
}
return path;
}
}
|
总结
以上就是本文关于Java二叉树路径和代码示例的全部内容,希望对大家有所帮助。感兴趣的朋友可以继续参阅本站其他相关专题,如有不足之处,欢迎留言指出。感谢朋友们对本站的支持!
原文链接:http://blog.csdn.net/qq_35440322/article/details/77850364