【题解】【剑指 Offer】34. 二叉树中和为某一值的路径

题目链接

 力扣

题目描述

给你二叉树的根节点 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;
    }
};