注:
if(root->left==nullptr&&root->right==nullptr&&count==target){
result.push_back(nums);
//这里不可以加return,不然会导致少进行一个nums.pop_back()
}
题目:
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:
给定如下二叉树,以及目标和 target = 22。
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
提示:
节点总数 <= 10000
题解:
从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
算法概述:
利用树的递归性质, 使用先序遍历, 将遍历经过的节点的 value 记录在数组 nums 中, 当到达叶子接节点后, if(sum(nums) == target) 则将 nums 记录在一个全局变量中 result, 遍历完整棵树后, result 即为解
递归的过程
首先使用先序遍历, 先序遍历是第一次访问到相应节点, 就进行操作的一种遍历方式, 回溯算法的题目基本都是采用先序遍历的方式。
从 root 节点开始, 当访问到一个节点时, 将节点的 value 记录在 nums 中, 如下图, 红色 的数组即为将相应节点的 value 记录到 nums 之后 nums 中的元素。
因为要记录所有经过的节点, 所以 nums 是要在递归过程中逐层的转递的, 所以当访问完某一个节点后,要将该节点的 value 从 nums 中删除。 如下图, 绿色 的数组, 即为将 value 从 nums 中删除的结果。
nums 中的元素, 通过上面两个步骤的操作, nums 中的元素始终都是代表从跟,到当前节点路径的所有元素
当到达任意叶子节点, 则计算 sum(nums) == target 如果相等, 则放入全局的 result 数组中
复杂度分析:
时间复杂度 O(N) : NN 为二叉树的节点数,先序遍历需要遍历所有节点。
空间复杂度 O(N) : 最差情况下,即树退化为链表时,path 存储所有树节点,使用 O(N)额外空间。
class Solution {
public:
vector<vector<int>> result;
vector<int> nums;
void dfs(TreeNode* root,int target,int count){
if(root==nullptr){
return ;
}
nums.push_back(root->val);
count=count+root->val;
if(root->left==nullptr&&root->right==nullptr&&count==target){
result.push_back(nums);
//这里不可以加return,不然会导致少进行一个nums.pop_back()
}
dfs(root->left,target,count);
dfs(root->right,target,count);
nums.pop_back();
}
vector<vector<int>> pathSum(TreeNode* root, int target) {
if(root==nullptr){
return {};
}
int count=0;
dfs(root,target,count);
return result;
}
};