路径总和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;
    }
};
N^2
复制代码

  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】

posted @   Renhr  阅读(21)  评论(0编辑  收藏  举报
编辑推荐:
· AI与.NET技术实操系列:基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
阅读排行:
· 25岁的心里话
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· 闲置电脑爆改个人服务器(超详细) #公网映射 #Vmware虚拟网络编辑器
· 一起来玩mcp_server_sqlite,让AI帮你做增删改查!!
· 零经验选手,Compose 一天开发一款小游戏!
点击右上角即可分享
微信分享提示