結果

問題 No.1197 モンスターショー
ユーザー e869120
提出日時 2021-03-05 21:02:43
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,404 bytes
コンパイル時間 753 ms
コンパイル使用メモリ 75,180 KB
実行使用メモリ 54,272 KB
最終ジャッジ日時 2024-10-06 23:40:47
合計ジャッジ時間 41,619 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 7 WA * 33 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
// 入力
int N, A[1 << 18], B[1 << 18];
int K, C[1 << 18];
int Q, T[1 << 18], X[1 << 18], Y[1 << 18];
// グラフ
vector<int> G[1 << 18], H[1 << 18];
bool used[1 << 18];
int cl[1 << 18], cr[1 << 18], dist[1 << 18], cnts;
// データ構造
int sz[1 << 18];
int dat[1 << 18][22];
// 出力
long long Answer[1 << 18];
void init() {
    for (int i = 0; i < 17; i++) {
        for (int j = (1 << i); j < (2 << i); j++) sz[j] = i;
    }
    for (int i = 0; i < 21; i++) {
        for (int j = 0; j <= 2 * N; j++) {
            int t = j + (1 << i);
            dat[i + 1][j] = dat[i][j];
            if (t <= 2 * N) dat[i + 1][j] = min(dat[i + 1][j], dat[i][t]);
        }
    }
}
int query(int cl, int cr) {
    // [cl, cr)
    if (cl >= cr) return (1 << 30);
    int f = sz[cr - cl];
    int v1 = dat[cl][f];
    int v2 = dat[cr - (1 << f)][f];
    return min(v1, v2);
}
void dfs(int pos, int dep) {
    cnts++; cl[pos] = cnts; dist[cnts] = dep; dat[cnts][0] = dep;
    used[pos] = true;
    for (int i = 0; i < G[pos].size(); i++) {
        if (used[G[pos][i]] == true) continue;
        H[pos].push_back(G[pos][i]);
        dfs(G[pos][i], dep + 1);
    }
    cnts++; cr[pos] = cnts; dist[cnts] = dep; dat[cnts][0] = dep;
}
int getdst(int u, int v) {
    u = cl[u];
    v = cl[v]; if (u > v) swap(u, v);
    return dist[u] + dist[v] - 2 * query(u, v + 1);
}
// クエリ平方分割
bool visited[1 << 18];
long long cnt[1 << 18];
long long dp[1 << 18], dp1[1 << 18], dp2[1 << 18];
void dfs1(int pos) {
    dp[pos] = cnt[pos];
    for (int i = 0; i < H[pos].size(); i++) {
        dfs1(H[pos][i]);
        dp[pos] += dp[H[pos][i]];
        dp1[pos] += dp1[H[pos][i]];
        dp1[pos] += dp[H[pos][i]];
    }
}
void dfs2(int pos) {
    long long r = 0;
    for (int i = 0; i < H[pos].size(); i++) {
        r += dp1[H[pos][i]] + dp[H[pos][i]];
    }
    for (int i = 0; i < H[pos].size(); i++) {
        long long s = r - (dp1[H[pos][i]] + dp[H[pos][i]]);
        long long t = dp[1] - dp[H[pos][i]];
        dp2[H[pos][i]] = dp2[pos] + (s + t);
        dfs2(H[pos][i]);
    }
}
void solve(int cl, int cr) {
    // Step #1. 初期化・前計算
    vector<int> vec;
    for (int i = 1; i <= K; i++) visited[i] = false;
    for (int i = 1; i <= N; i++) cnt[i] = 0;
    for (int i = 1; i <= N; i++) dp[i] = 0;
    for (int i = 1; i <= N; i++) dp1[i] = 0;
    for (int i = 1; i <= N; i++) dp2[i] = 0;
    for (int i = cl; i <= cr; i++) {
        if (T[i] == 1) visited[X[i]] = true;
    }
    for (int i = 1; i <= K; i++) {
        if (visited[i] == true) vec.push_back(i);
        else cnt[C[i]] += 1;
    }
    // Step #2. 全方位木 DP
    dfs1(1);
    dfs2(1);
    // Step #3. クエリ平方分割
    for (int i = cl; i <= cr; i++) {
        if (T[i] == 1) {
            C[X[i]] = Y[i];
        }
        if (T[i] == 2) {
            Answer[i] = dp1[X[i]] + dp2[X[i]];
            for (int j : vec) {
                Answer[i] += getdst(X[i], C[j]);
            }
        }
    }
}
int main() {
    // Step #1. 入力
    cin >> N >> K >> Q;
    for (int i = 1; i <= K; i++) cin >> C[i];
    for (int i = 1; i <= N - 1; i++) {
        cin >> A[i] >> B[i];
        G[A[i]].push_back(B[i]);
        G[B[i]].push_back(A[i]);
    }
    for (int i = 1; i <= Q; i++) {
        cin >> T[i];
        if (T[i] == 1) cin >> X[i] >> Y[i];
        if (T[i] == 2) cin >> X[i];
    }
    // Step #2. オイラーツアー
    dfs(1, 0);
    init();
    // Step #3. クエリ平方分割
    int Backet = 300;
    for (int i = 1; i <= Q; i += Backet) {
        int cl = i, cr = min(i + Backet - 1, Q);
        solve(cl, cr);
    }
    // Step #4. 出力
    for (int i = 1; i <= Q; i++) {
        if (T[i] == 2) cout << Answer[i] << endl;
    }
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0