父子情深

思路

可将数轴上求前缀和的思路变换一下,转成在有根树上从根结点向叶结点求树上前缀和。
时间复杂度

Code

/**
 * struct Point {
 *    int x;
 *    int y;
 * };
 */

class Solution {
private:
    #define maxn 100010
    vector<int> g[maxn];
    vector<long> val;
public:
    /**
     * 从 1 到 n 每个结点的权值。
     * @param n int整型
     * @param Edge Point类vector (u, v) 集合
     * @param q int整型
     * @param Query Point类vector Point.x 为题中 r, Point.y题中 为 x
     * @return long长整型vector
     */
    void dfs(int x, int fa = 0) {
        val[x] += val[fa];
        for (auto v: g[x]) {
            if (v == fa) continue;
            dfs(v, x);
        }
    }
    vector<long> solve(int n, vector<Point>& Edge, int q, vector<Point>& Query) {
        // write code here
        val.resize(n + 1);
        for (int i = 1; i <= n; ++i) {
            g[i].clear();
            val[i] = 0;
        }
        for (auto cur: Edge) {
            g[cur.x].push_back(cur.y);
            g[cur.y].push_back(cur.x);
        }
        for (int i = 0; i < q; ++i) {
            val[Query[i].x] += Query[i].y;
        }
        dfs(1);
        return {val.begin() + 1, val.end()};
    }
};
全部评论

相关推荐

研一开学九月份速成的Java,项目是苍穹外卖和黑马点评,算法基础不好,八股文较为熟练,想找份小厂日常实习,希望牛友们给点意见,蟹蟹啦
求offer的花生米很聪敏:三个月学了这么多?spring springmvc mybatis springboot jvm juc,还做完了两个项目,还熟悉八股,会点算法。卧槽,我该反思了。我暑假开始的,就做了外卖,spring springmvc boot 那些原理好多都忘了,还在刷 jvm 视频,八股和算法也没开始
点赞 评论 收藏
分享
11-08 17:36
诺瓦科技_HR
点赞 评论 收藏
分享
11-15 18:39
已编辑
西安交通大学 Java
全村最靓的仔仔:卧槽,佬啥bg呢,本也是西交么
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务