113. Path Sum II
仅供自己学习
思路:
我们通过DFS递归把所有的分支到叶子节点的都计算看是不是满足这个目标值。
首先判断递归传入的root是否是NULL,如果是就返回,不是就把这个sum-root->val,然后加入进path中去,如果 sum=0说明这条路是满足了target,那么久加入进res中,加入后不直接return,虽然已经可以结束此次递归,但我们还要将这个root删除掉,否则就会一直留有这个分支的这个元素到遍历另外分支,并且为了避免没有到叶子节点就让sum=0的情况,还需要判断root左右子树是否为NULL。然后再分别递归左右子树,最后在pop掉这个root即可。因为要遍历左右子树,那么需要弹出一遍后才遍历另一边,就是反悔,再找一条满足条件的路。
代码:
/**
* 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 {
private:
vector<vector<int>> res;
vector<int> path;
public:
void backtracking(TreeNode* root,int sum){
if(root==NULL) return ;
path.emplace_back(root->val);
sum-=root->val;
if(root->left ==NULL && root->right == NULL &&sum==0) {
res.push_back(path);
}
backtracking(root->left,sum);
backtracking(root->right,sum);
path.pop_back();
}
vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
backtracking(root,targetSum);
return res;
}
};