题目链接
题目描述
给你二叉树的根节点 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
输出:[]
解题思路
思路是没什么问题,就是判断的时候需要注意一下
void dfs(TreeNode* root,int sum,int target){
if(root==NULL) return;
sum+=root->val;
temp.push_back(root->val);
if(sum==target){
if(root->left==NULL&&root->right==NULL) v.push_back(temp);
//不能return
}
dfs(root->left,sum,target);
dfs(root->right,sum,target);
temp.pop_back();
sum-=root->val;
}
当判断是否相等的时候,此时相等的话不能直接return回去,因为每个节点的值可能是正可能是负,例如当输入
[1,-2,-3,1,3,-2,null,-1]
-1
此时的树结构为
答案应该去在红色箭头处,但是如果return,就会在蓝色箭头处不遍历下面的子树,就导致结果错误
题解
class Solution {
public:
vector<vector<int>> v;
vector<int> temp;
void dfs(TreeNode* root,int sum,int target){
if(root==NULL) return;
sum+=root->val;
temp.push_back(root->val);
if(sum==target){
if(root->left==NULL&&root->right==NULL) v.push_back(temp);
}
dfs(root->left,sum,target);
dfs(root->right,sum,target);
temp.pop_back();
sum-=root->val;
}
vector<vector<int>> pathSum(TreeNode* root, int target) {
if(root==NULL) return v;
dfs(root,0,target);
return v;
}
};