路径总和III

原题在这里：

　　概述题意，给定二叉树和一个targetsum值，问在二叉树中连续(父子节点)片段，其和为targetsum的数量。

节点定义：

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) {}
};

 

analyse：

　　1.mine，前缀和数组深搜，时间复杂度n^2

code:

class Solution
{
    int ans, target;
    void find(TreeNode *n, vector<long long> sum)
    {
        if (!n)
            return;
        for (long long &i : sum)
        {
            i += n->val;
            if (i == target)
                ++ans;
        }
        sum.emplace_back(n->val);
        if (n->val == target)
            ++ans;
        find(n->left, sum);
        find(n->right, sum);
    }

public:
    int pathSum(TreeNode *root, int targetSum)
    {
        target = targetSum;
        find(root, vector<long long>(0));
        return ans;
    }
};

　　2.还是前缀和，考虑 当前节点前缀和now 和 历史前缀和map：

　　　　若存在差值为targetsum，即表示当前节点到根节点的某一个节点之间片段能构成targetsum。

code：

class Solution
{
    unordered_map<long long, int> mp;//存入前缀和
    int find(TreeNode *n, long long now, int sum)
    {
        if (!n)
            return 0;
        int ans = 0;
        now += n->val;
        if (mp.count(now - sum))//看当前前缀和与往前前缀和之差(即为当前sum构成部分)是否存在
            ans = mp[now - sum];
        ++mp[now];
        ans += find(n->left, now, sum);
        ans += find(n->right, now, sum);
        --mp[now];
        return ans;
    }

public:
    int pathSum(TreeNode *root, int targetSum)
    {
        mp[0] = 1;
        return find(root, 0, targetSum);
    }
};

 

【Over】