给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
分析:
要找从根节点到叶子节点的数值和等于目标值,难免要遍历所有节点,使用深度优先遍历(DFS),从头遍历到为,只要达到目标值就,存储为数组,遍历结束,即可返回结果
程序中使用了sum累减直至等于0区判断,也可以使用累加,达到目标值
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {
if(root==nullptr)return res;
recur(root, sum);
return res;
}
private:
vector<vector<int>> res;
vector<int> path;
void recur(TreeNode* root, int sum)
{
if(root==nullptr){
return ;
}
path.push_back(root->val);
sum= sum-root->val;
if(sum ==0&&root->left==nullptr&&root->right==nullptr)
{ //从根节点到叶子节点 路径总和等于给定目标和的路径 必须有左右叶子节点走完
res.push_back(path);
}
recur(root->left,sum);
recur(root->right,sum);
path.pop_back();
}
};
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {
if(root==nullptr)return res;
recur(root, sum , 0);
return res;
}
private:
vector<vector<int>> res;
vector<int> path;
void recur(TreeNode* root, int sum,int k)
{
if(root==nullptr){
return ;
}
path.push_back(root->val);
k+=root->val;
if(sum ==k&&root->left==nullptr&&root->right==nullptr)
{ //从根节点到叶子节点 路径总和等于给定目标和的路径 必须有左右叶子节点走完
res.push_back(path);
}
recur(root->left,sum,k);
recur(root->right,sum,k);
path.pop_back();
}
};