给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
示例 1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]
示例 2:
输入:root = [1,2,3], targetSum = 5
输出:[]
示例 3:输入:root = [1,2], targetSum = 0
输出:[]来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
对于这道题,我们需要采用回溯的方法,使用深度优先遍历二叉树。使用一个辅助容器tmp来记录我们这一路上的所有经过的元素,并且用target减去我们每一次入栈的数据,表示我们还需要多少才能够匹配上我们的target。
无论我们当前的路径匹配成功与否,我们在匹配下一个路径的时候我们都需要将当前结点的元素也就是我们tmp中的最后一个元素删掉,然后再进入下一个分支进行计算(回溯)。
/**
* 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<int> tmp;
vector<vector<int>> result;
vector<vector<int>> pathSum(TreeNode* root, int target) {
sum(root,target);
return result;
}
void sum(TreeNode*root,int target)
{
if(root==nullptr)
{
return;
}
tmp.push_back(root->val);
target=target-root->val;
if(root->left==nullptr&&root->right==nullptr&&target==0)
{
result.push_back(tmp);
}
sum(root->left,target);
sum(root->right,target);
tmp.pop_back();
}
};