题目描述
给你二叉树的根节点
r
o
o
t
root
root 和一个整数目标和
t
a
r
g
e
t
S
u
m
targetSum
targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
方法一:回溯法
本问题是典型的二叉树方案搜索问题,使用回溯法解决,其包含 先序遍历 + 路径记录 两部分。
- 先序遍历: 按照 “根、左、右” 的顺序,遍历树的所有节点。
- 路径记录: 在先序遍历中,记录从根节点到当前节点的路径。当路径为 ① ① ①根节点到叶节点形成的路径 且 ② ② ② 各节点值的和等于目标值 s u m sum sum 时,将此路径加入结果列表。
算法流程:
pathSum(root, sum) 函数:
- 初始化:结果列表 r e s res res ,路径列表 p a t h path path 。
- 返回值: 返回 r e s res res 即可。
recur(root, tar) 函数:
- 递推参数: 当前节点 r o o t root root ,当前目标值 t a r tar tar 。
- 终止条件: 若节点 r o o t root root 为空,则直接返回。
- 递推工作:
1.路径更新:将当前节点值 root.val 加入路径 path ;
2.目标值更新: tar = tar - root.val(即目标值 t a r tar tar 从 s u m sum sum 减至 0 0 0 );
3.路径记录: 当 ① r o o t ① root ①root 为叶节点 且 ② ② ② 路径和等于目标值 ,则将此路径 p a t h path path 加入 r e s res res 。
4.先序遍历:递归左/右子节点。
5.路径恢复:向上回溯前,需要将当前节点从路径 p a t h path path 中删除,即执行 path.pop() 。
复杂度分析
- 时间复杂度 O ( N ) O(N) O(N) : N N N 为二叉树的节点数,先序遍历需要遍历所有节点。
- 空间复杂度 O ( N ) O(N) O(N) : 最差情况下,即树退化为链表时, p a t h path path 存储所有树节点,使用 O ( N ) O(N) O(N) 额外空间。
C++代码实现
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
vector<vector<int>> pathSum(TreeNode* root, int target) {
helper(root, target);
return res;
}
private:
void helper(TreeNode* root , int target){
if(root == nullptr) return;
path.push_back(root->val);
target -= root->val;
if(target == 0 && root->left == nullptr && root->right == nullptr){
res.push_back(path);
}
helper(root->left ,target);
helper(root->right ,target);
path.pop_back();
}
};