题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
解题策略
先序dfs+回溯法 + 递归
解题思路参考Krahets
算法流程
FindPath(TreeNode root,int target)函数
- 参数:待判断的树,给定的和;
- 返回值: 返回所有路径
help(TreeNode root, int tar)
-
递归参数: 当前节点 root ,当前目标值 tar。
-
递归终止条件:若节点 root 为空,则直接返回。
-
递归部分:
- 路径更新: 将当前节点值 root.val 加入路径 path ;
- 目标值更新: tar = tar - root.val(即目标值 tar 从 sum 减至 00 );
- 路径记录: 当 ① root 为叶节点 且 ② 路径和等于目标值 ,则将此路径 path 加入 res。
- 先序遍历: 递归左 / 右子节点。
- 路径恢复: 向上回溯前,需要将当前节点从路径 path 中删除,即执行 path.pop()。
-
返回值:无。
复杂度分析:
时间复杂度 O(N) : N 为二叉树的节点数,先序遍历需要遍历所有节点。
空间复杂度 O(N) : 最差情况下,即树退化为链表时,path 存储所有树节点,使用O(N) 额外空间。
注意
-
记录路径时若直接执行 res.add(path) ,则是将 path 对象加入了 res ;后续 path 改变时, res 中的 path 对象也会随之改变。
正确做法:res.add(new ArrayList(path)),相当于复制了一个 path 并加入到 res 。 -
函数不要混淆了:求ArrayList的长度,是size(), 不是length()!!!
import java.util.ArrayList;
public class Solution {
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> path = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
help(root, target);
return res;
}
void help(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 ArrayList(path));
}
help(root.left, tar);
help(root.right, tar);
path.remove(path.size()-1);// ArrayList的长度,是size(), 不是length()!!!!
}
}