1.回溯法思想是对的,但起始写的回溯函数写的有问题,调试不出来:(下面是起始写的有问题的回溯代码)
void recur(TreeNode* root, int target, vector<int>tmp, int sum) {
if (root == NULL||sum>target) return;
if (sum == target) {
ans.push_back(tmp);
}
tmp.push_back(root->val);
sum += root->val;
recur(root->left, target, tmp, sum);
tmp.pop_back();
sum -= root->val;
recur(root->right, target, tmp, sum);
}
2.看了大佬的题解后修改的代码。参考:面试题34. 二叉树中和为某一值的路径(回溯法,清晰图解)
原来自己写的回溯函数代码存在的问题及改正:
1)需要先操作后判断,否则为空;
2)需要在函数的最后才移除tmp一维数组的尾元素,才能正确回溯;
3)返回的路径要求是叶子节点,审题不清楚。
出错的主要原因在于下面递推工作的第3点,需要先更新后记录。
递推工作:1)路径更新;2)目标值更新;3)路径记录;4)先序遍历;5)路径恢复。
/**
* 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>> ans;
vector<int>tmp;
void recur(TreeNode *root,int target,int sum){
if(root==NULL) return;
tmp.push_back(root->val);
sum+=root->val;
if(sum==target&&root->left==NULL&&root->right==NULL){
ans.push_back(tmp);
}
recur(root->left,target,sum);
recur(root->right,target,sum);
tmp.pop_back();
}
vector<vector<int>> pathSum(TreeNode* root, int target) {
recur(root,target,0);
return ans;
}
};