1. 题目描述
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
2. 思路
(1)先序遍历路径,并将遍历的结点加入路径;
(2) 如果遍历到路径最后一个节点时,路径的和等于目标值,则将路径路径加入结果,否则则需要回溯到上一个结点,移出路径的最后一个节点的值。
(3)先序遍历完所有路径,返回结果。
3. 代码
class Solution {
List<List<Integer>> res = new ArrayList<List<Integer>>();//用于存结果
List<Integer> temp = new ArrayList<Integer>();//用于存路径
int target;
public List<List<Integer>> pathSum(TreeNode root, int target) {
this.target = target;
int sum = 0;
inorder(root,sum);//先序遍历
return res;
}
public void inorder(TreeNode root,int sum){
if(root==null)//递归终止条件
return;
//计算到该节点时路径的和
sum += root.val;
//将当前节点的值加入路径
temp.add(root.val);
//符合条件,将路径存入结果
if(sum==target && root.left==null && root.right == null){
res.add(new ArrayList(temp));
}
inorder(root.left,sum);
inorder(root.right,sum);
//回溯,移出路径最后一个节点的值
temp.remove(temp.size()-1);
}
}