結果

問題 No.1038 TreeAddQuery
ユーザー potoooooooopotoooooooo
提出日時 2020-04-25 21:06:42
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,703 ms / 4,000 ms
コード長 3,504 bytes
コンパイル時間 2,824 ms
コンパイル使用メモリ 222,636 KB
実行使用メモリ 162,568 KB
最終ジャッジ日時 2024-05-03 23:20:11
合計ジャッジ時間 25,860 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
14,048 KB
testcase_01 AC 5 ms
13,972 KB
testcase_02 AC 4 ms
14,012 KB
testcase_03 AC 19 ms
16,068 KB
testcase_04 AC 19 ms
15,960 KB
testcase_05 AC 18 ms
16,228 KB
testcase_06 AC 16 ms
15,932 KB
testcase_07 AC 18 ms
16,192 KB
testcase_08 AC 1,092 ms
119,640 KB
testcase_09 AC 1,210 ms
125,176 KB
testcase_10 AC 1,241 ms
125,852 KB
testcase_11 AC 1,275 ms
125,432 KB
testcase_12 AC 1,252 ms
126,424 KB
testcase_13 AC 1,703 ms
162,568 KB
testcase_14 AC 1,541 ms
135,244 KB
testcase_15 AC 1,443 ms
132,444 KB
testcase_16 AC 1,443 ms
131,360 KB
testcase_17 AC 1,418 ms
130,268 KB
testcase_18 AC 387 ms
70,732 KB
testcase_19 AC 481 ms
75,908 KB
testcase_20 AC 411 ms
76,856 KB
testcase_21 AC 506 ms
77,956 KB
testcase_22 AC 714 ms
92,752 KB
testcase_23 AC 743 ms
94,584 KB
testcase_24 AC 1,048 ms
124,124 KB
testcase_25 AC 1,694 ms
162,556 KB
testcase_26 AC 1,075 ms
117,444 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
In member function 'Fenwick<long long int>& Fenwick<long long int>::operator=(Fenwick<long long int>&&)',
    inlined from 'Info& Info::operator=(Info&&)' at main.cpp:24:8,
    inlined from 'centroid(int)::<lambda(int, int)>' at main.cpp:67:30:
main.cpp:6:26: warning: '<unnamed>.Info::fw.Fenwick<long long int>::N' may be used uninitialized [-Wmaybe-uninitialized]
    6 | template<class T> struct Fenwick {
      |                          ^~~~~~~
main.cpp: In lambda function:
main.cpp:67:30: note: '<anonymous>' declared here
   67 |         F[curr][root] = Info();
      |                              ^

ソースコード

diff #

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

// zero-based numbering
template<class T> struct Fenwick {
    vector<T> bit; int N;
    Fenwick() {}
    Fenwick(int n) : N(n) { bit.assign(n+1, 0); }
    // add w to a
    void add(int a, T w) {
        for (int x = ++a; x <= N; x += x & -x) bit[x] += w;
    }
    // sum [0, a)
    T sum(int a) {
        T ret = 0;
        for (int x = a; x > 0; x -= x & -x) ret += bit[x];
        return ret;
    }
    // sum [a, b)
    T sum(int a, int b) { return sum(b) - sum(a); }
};

struct Info {
    vector<int> h;
    Fenwick<long long> fw;
    int max_h = 0;
    Info() : h{{0}} {}
    Info(int sz) : fw(sz+2), h{{0}} {};
};

vector<int> e[MAX_SIZE];
int sz[MAX_SIZE];
bool cut[MAX_SIZE];
unordered_map<int, Info> F[MAX_SIZE];
vector<array<int, 4>> v[MAX_SIZE]; 

int find_centroid(int curr) {
    auto dfs = [&](auto f, int curr, int prev) -> void {
        sz[curr] = 1;
        for (auto &to: e[curr]) {
            if (to == prev || cut[to]) continue;
            f(f, to, curr);
            sz[curr] += sz[to];
        }
    };
    dfs(dfs, curr, -1);
    int all = sz[curr];
    while (true) {
        int next = -1;
        for (auto &to: e[curr]) {
            if (cut[to] || sz[to] > sz[curr]) continue;
            if (sz[to] * 2 >= all) next = to;
        }
        if (next == -1) break;
        curr = next;
    }
    cut[curr] = true;
    return curr;
}

long long cntt = 0;

void centroid(int curr) {
    curr = find_centroid(curr);
    auto bfs = [&](int root, int prev) {
        F[curr][root] = Info();
        cntt += sz[root];
        auto &u = F[curr][root];
        queue<array<int, 3>> q;
        q.push({root, prev, 0});
        int cnt = 0;
        while (!q.empty()) {
            auto c = q.front(); q.pop();
            v[c[0]].push_back({curr, root, cnt, c[2]});
            if (u.max_h < c[2]) {
                u.max_h = c[2];
                u.h.emplace_back(0);
            }
            u.h.back() = cnt++;
            for (auto &x: e[c[0]]) {
                if (x == c[1] || cut[x]) continue;
                q.push({x, c[0], c[2]+1});
            }
        }
        F[curr][root].fw = Fenwick<long long>(cnt);
    };
    bfs(curr, -1);
    for (auto &to: e[curr]) {
        if (cut[to]) continue;
        bfs(to, curr);
    }
    for (auto &to: e[curr]) {
        if (!cut[to]) centroid(to);
    }
}

long long f(int x) {
    long long ans = 0;
    for (auto &[c, r, i, h] : v[x]) {
        ans += F[c][r].fw.sum(i+1);
    }
    return ans;
}

long long query(int x, int y, long long z) {
    long long ans = 0;
    for (auto &[c, r, i, h] : v[x]) {
        ans += F[c][r].fw.sum(i+1);
        int d = (c == r) ? y-h : y-h-2;
        d = min(d, F[c][r].max_h);
        if (d >= 0) {
            long long w = (c == r) ? z : -z;
            F[c][r].fw.add(0, w);
            F[c][r].fw.add(F[c][r].h[d]+1, -w);
        }
    }
    return ans;
}

int main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    int n, q; cin >> n >> q;
    for (int i = 0; i < n-1; i++) {
        int u, v; cin >> u >> v; u--; v--;
        e[u].emplace_back(v);
        e[v].emplace_back(u);
    }
    centroid(0);
    
    auto debug = [&]() {
        for (int j = 0; j < n; j++) cout << f(j) << " ";
        cout << "\n";
    };

    for (int i = 0; i < q; i++) {
        int x, y; long long z;
        cin >> x >> y >> z;
        cout << query(--x, y, z) << "\n";
        // debug();
    }
    return 0;
}
0