给定一个二叉树和一个值sum,请找出所有的根节点到叶子节点的节点值之和等于sum的路径,
例如:
给出如下的二叉树,sum=22,
返回
[ [5,4,11,2], [5,8,4,5]]
根据path-sum这个的思路以及针对树问题的解法,首先考虑的就是递归,关键就是这个递归怎么构造。递归的思路很简单:就是递归遍历一棵树,也就是先序遍历,在遍历的过程中去把节点上的值收集起来,满足的留下,不满足的去掉
import java.util.ArrayList;
public class PathSumII {
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) {
val = x;
}
}
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
public ArrayList<ArrayList<Integer>> pathSum(TreeNode root, int sum) {
ArrayList<Integer> list = new ArrayList<Integer>();
process(root,sum,0,list);
return res;
}
private void process(TreeNode root, int sum,int res, ArrayList<Integer> list) {
// TODO Auto-generated method stub
if(root == null) {
return ;
}
if(root.left == null && root.right == null && res + root.val == sum) {
list.add(root.val);
this.res.add(new ArrayList<Integer>(list));
list.remove(list.size() - 1);
return ;
}
list.add(root.val);
process(root.left, sum, res+root.val, list);
process(root.right, sum, res+root.val, list);
list.remove(list.size() - 1);
}
}
注意:
代码中的两个细节,我直接截图牛客的评论区了,感觉人家问就是我当时困惑的,解释的让我直接就清楚了
真的强烈建议大家做完一道题后,在评论区里多看会,多学习一下人家的思路和一些细节上的处理。