結果

問題 No.1197 モンスターショー
ユーザー 沙耶花沙耶花
提出日時 2020-08-22 17:47:10
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,325 ms / 3,000 ms
コード長 3,212 bytes
コンパイル時間 2,260 ms
コンパイル使用メモリ 217,528 KB
実行使用メモリ 54,292 KB
最終ジャッジ日時 2024-10-15 18:35:51
合計ジャッジ時間 29,382 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 212 ms
45,560 KB
testcase_08 AC 146 ms
19,628 KB
testcase_09 AC 522 ms
33,496 KB
testcase_10 AC 1,149 ms
32,456 KB
testcase_11 AC 451 ms
12,464 KB
testcase_12 AC 209 ms
7,936 KB
testcase_13 AC 202 ms
22,200 KB
testcase_14 AC 721 ms
22,596 KB
testcase_15 AC 438 ms
11,768 KB
testcase_16 AC 1,085 ms
38,832 KB
testcase_17 AC 1,010 ms
39,292 KB
testcase_18 AC 330 ms
14,460 KB
testcase_19 AC 985 ms
32,392 KB
testcase_20 AC 122 ms
12,212 KB
testcase_21 AC 383 ms
5,504 KB
testcase_22 AC 29 ms
5,376 KB
testcase_23 AC 981 ms
24,504 KB
testcase_24 AC 563 ms
24,196 KB
testcase_25 AC 248 ms
16,984 KB
testcase_26 AC 533 ms
17,788 KB
testcase_27 AC 679 ms
35,732 KB
testcase_28 AC 652 ms
19,768 KB
testcase_29 AC 339 ms
23,316 KB
testcase_30 AC 223 ms
26,700 KB
testcase_31 AC 333 ms
21,112 KB
testcase_32 AC 323 ms
5,248 KB
testcase_33 AC 180 ms
17,036 KB
testcase_34 AC 1,546 ms
41,232 KB
testcase_35 AC 1,516 ms
41,264 KB
testcase_36 AC 1,533 ms
41,400 KB
testcase_37 AC 1,531 ms
41,264 KB
testcase_38 AC 1,568 ms
41,404 KB
testcase_39 AC 1,540 ms
41,528 KB
testcase_40 AC 2,325 ms
54,292 KB
testcase_41 AC 1 ms
5,248 KB
testcase_42 AC 2 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define modulo 998244353
#define mod(mod_x) ((((long long)mod_x+modulo))%modulo)
#define Inf 1000000005

vector<pair<int,int>> order;
vector<int> L;
int N,K,Q;
vector<int> dis;
vector<long long> calc;
vector<int> cnt;
vector<int> C;

template <typename T>
struct sparse_table{
	const T init_value = make_pair(1000000001,1000001);
	
	vector<T> v;
	int n;
	int sz;
	sparse_table(vector<T> &x){
		sz = x.size();
		for(int i=0;true;i++){
			if((1<<i)>sz){
				n = i;
				break;
			}
		}
		v.resize(sz*n,init_value);
		
		for(int i=0;i<sz;i++){
			v[i] = x[i];
		}
		
		for(int i=1;i<n;i++){
			for(int j=0;j<sz+1-(1<<i);j++){
				int temp = j+(1<<(i-1));
				v[i*sz+j] = func(v[(i-1)*sz+j],v[(i-1)*sz+temp]);
			}
		}
	}
	
	T query(int l,int r){
		int x = 31-__builtin_clz(r-l);
		return func(v[sz*x+l],v[sz*x+r-(1<<x)]);
	}
	
	T func(T a,T b){
		return min(a,b);
	}
};

void dfs(vector<vector<int>> &E,int now,int p){
	cnt[now] = C[now];
	for(int i=0;i<E[now].size();i++){
		dfs(E,E[now][i],now);
		cnt[now] += cnt[E[now][i]];
	}
}

void dfs2(vector<vector<int>> &E,int now,int p){
	if(p==-1){
		calc[now] = 0;
		for(int i=0;i<N;i++){
			calc[now] += (long long)dis[i] * C[i];
		}
	}
	else{
		calc[now] = calc[p];
		calc[now] -= cnt[now];
		calc[now] += K-cnt[now];
	}

	for(int i=0;i<E[now].size();i++){
		dfs2(E,E[now][i],now);
	}
}

void Erasedfs(vector<vector<int>> &E,int now,int p,int d=0){
	dis[now] =d;
	L[now] = order.size();
	order.emplace_back(d,now);
	vector<int> t;
	for(int i=0;i<E[now].size();i++){
		if(E[now][i]==p)continue;
		t.push_back(E[now][i]);
		Erasedfs(E,E[now][i],now,d+1);
		order.emplace_back(d,now);
	}
	E[now] = t;
	
}

int get_distance(sparse_table<pair<int,int>> &S,int u,int v){
	if(u==v)return 0;
	int l = min(L[u],L[v]);
	int r = max(L[u],L[v]);
	int x = S.query(l,r+1).second;
	return dis[u]+dis[v]-2*dis[x];
}
	

int main(){
	
	cin>>N>>K>>Q;
	C.resize(N,0);
	vector<int> pos(K);
	for(int i=0;i<K;i++){
		scanf("%d",&pos[i]);
		pos[i]--;
		C[pos[i]]++;
	}
	
	vector<vector<int>> E(N);
	for(int i=0;i<N-1;i++){
		int a,b;
		scanf("%d %d",&a,&b);
		a--;b--;
		E[a].push_back(b);
		E[b].push_back(a);
	}
	dis.resize(N);
	L.resize(N);
	Erasedfs(E,0,-1);
	
	
	sparse_table<pair<int,int>> S(order);
	
	calc.resize(N);
	cnt.resize(N);
	dfs(E,0,-1);
	dfs2(E,0,-1);
	vector<pair<int,int>> Movement;
	for(int i=0;i<Q;i++){
		
		if(Movement.size()>350){
			for(int j=0;j<Movement.size();j++){
				C[pos[Movement[j].first]]--;
				pos[Movement[j].first] = Movement[j].second;
				C[pos[Movement[j].first]]++;
			}
			Movement.resize(0);
			dfs(E,0,-1);
			dfs2(E,0,-1);
		}
		
		int t;
		scanf("%d",&t);
		
		if(t==1){
			int p,d;
			scanf("%d %d",&p,&d);
			p--;d--;
			
			bool f = false;
			for(int j=0;j<Movement.size();j++){
				if(Movement[j].first==p){
					Movement[j].second = d;
					f=true;
					break;
				}
			}
			if(!f)Movement.emplace_back(p,d);
			
		}
		else{
			int e;
			scanf("%d",&e);
			e--;
			long long ans = calc[e];
			for(int j=0;j<Movement.size();j++){
				ans -= get_distance(S,e,pos[Movement[j].first]);
				ans += get_distance(S,e,Movement[j].second);
			}
			
			printf("%lld\n",ans);
			
		}
	}
	
	return 0;
}
0