LeetCode-树-求和路径

给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。

示例:
给定如下二叉树，以及目标和 sum = 22，

       
返回:

3
解释：和为 22 的路径有：[5,4,11,2], [5,8,4,5], [4,11,7]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/paths-with-sum-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

思路：

树的先序遍历

先计算当前结点到之前的结点是否满足，返回个数

然后添加左右子节点，递归判断

注意，左右子节点判断满足条件的个数时，需要排除掉之前已经查找过的，否则会导致重复。

代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int pathSum(TreeNode* root, int sum) {
        int res=0;
        vector<int> path;
        PathSum(root,sum,res,path);
        return res;
    }


    void PathSum(TreeNode* node, int sum, int& res, vector<int>& path)
    {
        if(node == nullptr)
        {
            return;
        }

        //到达叶子结点判断个数
        if(node->left==nullptr && node->right==nullptr)
        {
            path.push_back(node->val);
            res += GetSumCount(path,sum);
            path.pop_back();
            return;
        }

        //添加路径，判断满足条件个数
        path.push_back(node->val);
        res += GetSumCount(path,sum);
        PathSum(node->left,sum,res,path);   //递归左子树
        PathSum(node->right,sum,res,path);  //递归右子树
        path.pop_back();
    }

    //计算新添加结点导致满足条件的个数
    int GetSumCount(const vector<int>& path,int aim)
    {
        int nRes=0;
        int sum=0;
        for(size_t i=0; i<path.size(); i++)
        {
            sum += path[i];
            //这里只判断新添加的结点，即最后一个结点，防止重复
            if(sum==aim && i==path.size()-1)
            {
                nRes++;
            }
        }

        //这里要排除的掉后一个新添加的结点
        for(size_t i=0; i<path.size()-1; i++)
        {
            sum -= path[i];
            if(sum == aim)
            {
                nRes++;
            }
        }
        return nRes;
    }
};