给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
思路:dfs+回溯
提交的代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
List<List<Integer>> result = new ArrayList<List<Integer>>();
List<Integer> list = new ArrayList<Integer>();
int count=0;
public void fun(TreeNode root, int sum)
{
if(count==sum&&root.left==null&&root.right==null)
{
result.add(new ArrayList<>(list)); //不能写result.add(list);因为list会改变,带动result也发生改变,需要复制一份到result
}
if(root.left!=null)
{
list.add(root.left.val);
count+=root.left.val;
fun(root.left,sum);
}
if(root.right!=null)
{
list.add(root.right.val);
count+=root.right.val;
fun(root.right,sum);
}
count-=list.get(list.size()-1);
list.remove(list.size()-1);
}
public List<List<Integer>> pathSum(TreeNode root, int sum) {
if(root!=null)
{
list.add(root.val);
count+=root.val;
fun(root,sum);
}
return result;
}
}