Java二叉树路径和代码示例

时间:2021-12-09 07:01:00

给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值 的路径。

一个有效的路径,指的是从根节点到叶节点的路径。

样例

给定一个二叉树,和 目标值 = 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