結果

問題 No.1197 モンスターショー
ユーザー e869120e869120
提出日時 2021-03-05 21:11:37
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,934 ms / 3,000 ms
コード長 3,421 bytes
コンパイル時間 804 ms
コンパイル使用メモリ 75,912 KB
実行使用メモリ 54,784 KB
最終ジャッジ日時 2024-04-16 08:31:50
合計ジャッジ時間 34,266 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 15 ms
16,896 KB
testcase_01 AC 13 ms
16,896 KB
testcase_02 AC 13 ms
16,896 KB
testcase_03 AC 13 ms
16,896 KB
testcase_04 AC 13 ms
17,024 KB
testcase_05 AC 12 ms
17,024 KB
testcase_06 AC 13 ms
17,024 KB
testcase_07 AC 368 ms
46,464 KB
testcase_08 AC 212 ms
29,824 KB
testcase_09 AC 670 ms
39,936 KB
testcase_10 AC 1,406 ms
40,192 KB
testcase_11 AC 579 ms
25,728 KB
testcase_12 AC 295 ms
21,760 KB
testcase_13 AC 273 ms
31,744 KB
testcase_14 AC 903 ms
33,152 KB
testcase_15 AC 558 ms
25,344 KB
testcase_16 AC 1,413 ms
44,416 KB
testcase_17 AC 1,279 ms
44,800 KB
testcase_18 AC 410 ms
26,752 KB
testcase_19 AC 1,263 ms
39,808 KB
testcase_20 AC 164 ms
24,448 KB
testcase_21 AC 473 ms
20,736 KB
testcase_22 AC 49 ms
18,688 KB
testcase_23 AC 1,190 ms
34,944 KB
testcase_24 AC 779 ms
33,920 KB
testcase_25 AC 326 ms
28,032 KB
testcase_26 AC 681 ms
29,184 KB
testcase_27 AC 926 ms
41,728 KB
testcase_28 AC 821 ms
31,104 KB
testcase_29 AC 454 ms
32,896 KB
testcase_30 AC 310 ms
35,072 KB
testcase_31 AC 437 ms
31,232 KB
testcase_32 AC 392 ms
19,584 KB
testcase_33 AC 237 ms
27,904 KB
testcase_34 AC 1,984 ms
46,848 KB
testcase_35 AC 1,840 ms
47,104 KB
testcase_36 AC 1,951 ms
46,848 KB
testcase_37 AC 1,810 ms
47,104 KB
testcase_38 AC 1,937 ms
46,848 KB
testcase_39 AC 1,908 ms
46,976 KB
testcase_40 AC 2,934 ms
54,784 KB
testcase_41 AC 8 ms
17,024 KB
testcase_42 AC 8 ms
17,024 KB
権限があれば一括ダウンロードができます

ソースコード

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 < 18; 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[j][i + 1] = dat[j][i];
			if (t <= 2 * N) dat[j][i + 1] = min(dat[j][i + 1], dat[t][i]);
		}
	}
}

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 - 0; dat[cnts][0] = dep - 0;
	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 - 1; dat[cnts][0] = dep - 1;
}

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 = 1000;
	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;
}
0