結果

問題 No.399 動的な領主
ユーザー maru143maru143
提出日時 2021-05-02 14:24:40
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 112 ms / 2,000 ms
コード長 3,041 bytes
コンパイル時間 3,064 ms
コンパイル使用メモリ 216,056 KB
実行使用メモリ 35,824 KB
最終ジャッジ日時 2024-07-20 23:17:19
合計ジャッジ時間 4,661 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 3 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 9 ms
5,376 KB
testcase_06 AC 108 ms
12,752 KB
testcase_07 AC 112 ms
12,628 KB
testcase_08 AC 108 ms
12,640 KB
testcase_09 AC 101 ms
12,612 KB
testcase_10 AC 3 ms
5,376 KB
testcase_11 AC 8 ms
5,376 KB
testcase_12 AC 73 ms
12,964 KB
testcase_13 AC 74 ms
12,944 KB
testcase_14 AC 75 ms
35,824 KB
testcase_15 AC 78 ms
35,820 KB
testcase_16 AC 66 ms
24,036 KB
testcase_17 AC 107 ms
12,568 KB
testcase_18 AC 109 ms
12,612 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

struct HeavyLightDecomposition {
    vector<int> sz, par, depth;
    vector<int> idx, revidx, hld, top;
    vector<vector<int>> G;
    int root;
    bool is_built = false;

    HeavyLightDecomposition(int n, int root = 0)
        : sz(n), par(n), depth(n), idx(n), revidx(n), top(n), G(n), root(root) {}

    void add_edge(int u, int v) {
        G[u].push_back(v);
        G[v].push_back(u);
    }

    void build() {
        dfs(root);
        dfs2(root);
        is_built = true;
    }

    void dfs(int v, int p = -1) {
        sz[v] = 1;
        par[v] = p;
        if (p >= 0) depth[v] = depth[p] + 1;
        for (int &to : G[v]) {
            if (to == p) continue;
            dfs(to, v);
            sz[v] += sz[to];
        }
    }

    void dfs2(int v, int p = -1) {
        idx[v] = hld.size();
        revidx[hld.size()] = v;
        hld.push_back(v);

        int x = 0;
        int heaviest = -1;
        for (int &to : G[v]) {
            if (to == p) continue;
            if (x < sz[to]) {
                x = sz[to];
                heaviest = to;
            }
        }

        if (heaviest < 0) return;

        top[heaviest] = top[v];
        dfs2(heaviest, v);
        for (int &to : G[v]) {
            if (to == heaviest || to == p) continue;
            top[to] = to;
            dfs2(to, v);
        }
    }

    // return [l, r)s
    vector<pair<int, int>> query(int u, int v) {
        assert(is_built);
        vector<pair<int, int>> ret;
        while (top[u] != top[v]) {
            if (depth[top[u]] > depth[top[v]]) swap(u, v);
            ret.push_back({idx[top[v]], idx[v] + 1});
            v = par[top[v]];
        }
        if (idx[u] > idx[v]) swap(u, v);
        ret.push_back({idx[u], idx[v] + 1});
        return ret;
    }

    i64 lca(int u, int v) {
        assert(is_built);
        while (top[u] != top[v]) {
            if (depth[top[u]] > depth[top[v]]) swap(u, v);
            v = par[top[v]];
        }
        if (idx[u] > idx[v]) swap(u, v);
        return u;
    }
};

void solve() {
    int n;
    cin >> n;
    HeavyLightDecomposition hld(n);
    for (int i = 0; i < n - 1; i++) {
        int a, b;
        cin >> a >> b;
        a--;
        b--;
        hld.add_edge(a, b);
    }
    hld.build();

    int q;
    cin >> q;
    vector<i64> dp(n);
    while (q--) {
        int a, b;
        cin >> a >> b;
        a--;
        b--;
        int lca = hld.lca(a, b);
        dp[a]++;
        dp[b]++;
        dp[lca]--;
        if (hld.par[lca] >= 0) dp[hld.par[lca]]--;
    }

    auto dfs = [&](auto self, int v) -> void {
        for (int &to : hld.G[v]) {
            if (to == hld.par[v]) continue;
            self(self, to);
            dp[v] += dp[to];
        }
    };

    dfs(dfs, 0);

    i64 ans = 0;
    for (int i = 0; i < n; i++) {
        ans += dp[i] * (dp[i] + 1) / 2;
    }
    cout << ans << endl;
}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    solve();
    return 0;
}
0