题目:输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
思路:先序遍历树,用全局变量LinkedList类型的 path 来记录路径,暴力回溯,tar 不停的递减,直到 tar 为 0 并且当前结点为叶子结点,那么找到正确的路径 path,将其存入结果集中。
- 时间复杂度O(n),遍历书中每个结点的时间复杂度
- 空间复杂度O(n),需要定义一个辅助数组 path ,来存储路径,结果集不算再空间复杂度里面。
class Solution {
// 定义两个全局变量比较好存储并且返回,不用通过传参或者函数返回值来修改
public LinkedList<List<Integer>> res = new LinkedList<>();
public LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
recur(root, sum);
return res;
}
// 不要写成 static ,不然里面定义的成员都是 static 的,全局变量的多个测试用例就会发生错误
void recur(TreeNode root, int tar) {
if (root == null) {
return ;
}
path.add(root.val);
tar -= root.val;
if (tar == 0 && root.left == null && root.right == null) {
res.add(new LinkedList(path));
}
recur(root.left, tar);
recur(root.right, tar);
// 因为 path 是全局变量,当前结点遍历完了,一定要进行回溯,
path.removeLast();
return ;
}
}