给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值 的路径。
一个有效的路径,指的是从根节点到叶节点的路径。
样例
给定一个二叉树,和 目标值 = 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