結果

問題 No.1197 モンスターショー
ユーザー e869120e869120
提出日時 2021-03-05 21:02:43
言語 C++14
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 6 ms
18,372 KB
testcase_01 AC 6 ms
18,372 KB
testcase_02 AC 8 ms
18,504 KB
testcase_03 AC 6 ms
18,376 KB
testcase_04 AC 6 ms
18,372 KB
testcase_05 AC 5 ms
18,372 KB
testcase_06 AC 5 ms
18,372 KB
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 TLE -
testcase_41 AC 11 ms
16,384 KB
testcase_42 AC 10 ms
16,512 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 < 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;
}
0