【每日一题】ABC138D - Ki | 搜索 | 简单

题目内容

原题链接

给定一个 $n$ 个点的有根树，$Q$ 次操作，每次操作将点 $p$ 的子树中所有点都加 $x$ ，问操作后每个点的权值是多少，初始每个点的权值都是 $0$

数据范围

• $2\le n,Q\le 2\cdot 10^5$
• $1\le p\le n$
• $1\le x\le 10^4$

题解

因为是有根树，所以按照从根到每个叶子的方式遍历，每个儿子累加上其父亲的权值即可。

时间复杂度：$O(n)$

代码

#include <bits/stdc++.h>
using namespace std;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, Q;
    cin >> n >> Q;

    vector<vector<int>> g(n);
    for (int i = 1; i < n; ++i) {
        int a, b;
        cin >> a >> b;
        a--, b--;
        g[a].push_back(b);
        g[b].push_back(a);
    }

    vector<int> val(n, -1);
    vector<int> add(n, 0);

    for (int i = 0; i < Q; ++i) {
        int p, x;
        cin >> p >> x;
        p--;
        add[p] += x;
    }

    queue<int> q;
    val[0] = add[0];
    q.push(0);

    while (!q.empty()) {
        int u = q.front(); q.pop();
        for (int v: g[u]) {
            if (val[v] == -1) {
                val[v] = val[u] + add[v];
                q.push(v);
            }
        }
    }

    for (int i = 0; i < n; ++i) cout << val[i] << " \n"[i + 1 == n];

    return 0;
}