题目
思路
显然这是一道DFS+回溯题,在DFS中加回溯需要在递归函数中维护一个外部变量,该变量用于记录递归信息,对该变量的回溯一般放在DFS函数的尾部,在到达递归的最深处开始回溯外部变量。
DFS与回溯代码模板:
type function(){
vector<vector<int>> res; //用于记录结果
vector<int> path; // 用于记录递归信息
if(正常运行的条件){
dfs(一般传入数、vector等,传入记录递归的变量,传入记录结果的变量);
}
return res;
}
void dfd(参数){
if(出递归的条件) return;
//执行的操作,对当前节点需要做的事情,非必须
//对记录递归的变量要做的事情,path.push_back(节点信息);
if(判断到当前节点为止,是否符合题目要求){
符合题目要求就按要求把数据放进res中。
}
dfs(下一个节点);//有时需要对下一个节点进行判断,看是否满足限制要求
path.pop_back(); //回溯,最重要的一步,在递归完成,退出当前节点的时候,需要把记录递归的变量进行回溯
}
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int target) {
vector<vector<int>> res;
vector<int> path;
if(root){
dfs(root,res,path,target);
}
return res;
}
void dfs(TreeNode* root,vector<vector<int>>& res, vector<int>& path,int target){
if(root==nullptr) return;
target-=root->val;
path.push_back(root->val);
if(target==0&&root->left==nullptr&&root->right==nullptr){
res.push_back(path);
}
dfs(root->left,res,path,target);
dfs(root->right,res,path,target);
path.pop_back();
}
};