結果
問題 | No.1197 モンスターショー |
ユーザー | autumn-eel |
提出日時 | 2020-10-10 16:06:06 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 529 ms / 3,000 ms |
コード長 | 2,229 bytes |
コンパイル時間 | 1,642 ms |
コンパイル使用メモリ | 172,172 KB |
実行使用メモリ | 57,792 KB |
最終ジャッジ日時 | 2023-09-27 21:31:40 |
合計ジャッジ時間 | 9,843 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 7 ms
29,404 KB |
testcase_01 | AC | 7 ms
29,324 KB |
testcase_02 | AC | 7 ms
29,312 KB |
testcase_03 | AC | 8 ms
31,356 KB |
testcase_04 | AC | 9 ms
31,532 KB |
testcase_05 | AC | 8 ms
31,468 KB |
testcase_06 | AC | 8 ms
31,284 KB |
testcase_07 | AC | 122 ms
54,620 KB |
testcase_08 | AC | 80 ms
43,568 KB |
testcase_09 | AC | 137 ms
48,136 KB |
testcase_10 | AC | 155 ms
46,108 KB |
testcase_11 | AC | 72 ms
41,872 KB |
testcase_12 | AC | 50 ms
39,000 KB |
testcase_13 | AC | 78 ms
43,776 KB |
testcase_14 | AC | 124 ms
44,048 KB |
testcase_15 | AC | 73 ms
41,872 KB |
testcase_16 | AC | 183 ms
47,316 KB |
testcase_17 | AC | 197 ms
49,624 KB |
testcase_18 | AC | 66 ms
42,224 KB |
testcase_19 | AC | 155 ms
45,852 KB |
testcase_20 | AC | 43 ms
41,736 KB |
testcase_21 | AC | 58 ms
38,388 KB |
testcase_22 | AC | 18 ms
38,056 KB |
testcase_23 | AC | 135 ms
44,380 KB |
testcase_24 | AC | 111 ms
44,312 KB |
testcase_25 | AC | 68 ms
42,764 KB |
testcase_26 | AC | 98 ms
43,044 KB |
testcase_27 | AC | 173 ms
46,788 KB |
testcase_28 | AC | 102 ms
43,468 KB |
testcase_29 | AC | 100 ms
44,240 KB |
testcase_30 | AC | 105 ms
44,912 KB |
testcase_31 | AC | 85 ms
43,872 KB |
testcase_32 | AC | 40 ms
37,748 KB |
testcase_33 | AC | 60 ms
42,680 KB |
testcase_34 | AC | 242 ms
50,092 KB |
testcase_35 | AC | 230 ms
48,076 KB |
testcase_36 | AC | 247 ms
50,076 KB |
testcase_37 | AC | 240 ms
48,068 KB |
testcase_38 | AC | 238 ms
47,996 KB |
testcase_39 | AC | 256 ms
50,308 KB |
testcase_40 | AC | 529 ms
57,792 KB |
testcase_41 | AC | 9 ms
31,636 KB |
testcase_42 | AC | 9 ms
31,368 KB |
ソースコード
#include <bits/stdc++.h> #define rep(i,n)for(int i=0;i<(n);i++) using namespace std; typedef long long ll; typedef pair<ll,ll>P; int dist[22][200000]; vector<int>E[200000],G[200000]; bool used[200000]; int N; int centroid; int sz[200000]; int par[200000]; int id[200000]; int dep[200000]; ll sum1[200000],cnt1[200000]; vector<ll>sum2[200000],cnt2[200000]; void szdfs(int v,int p){ sz[v]=1; for(int u:E[v]){ if(u==p||used[u])continue; szdfs(u,v); sz[v]+=sz[u]; } } void search_centroid(int v,int p){ bool ok=true; for(int u:E[v]){ if(u==p||used[u])continue; search_centroid(u,v); if(sz[u]>N/2)ok=false; } if(N-sz[v]>N/2)ok=false; if(ok)centroid=v; } void solve(int v,int p){ szdfs(v,-1); N=sz[v]; search_centroid(v,-1); int s=centroid; used[s]=true; par[s]=p; if(p!=-1)G[p].push_back(s); for(int u:E[s]){ if(used[u])continue; solve(u,s); } } void build(int v,int p,int d,int D){ dist[D][v]=d; for(int u:E[v]){ if(used[u]||u==p)continue; build(u,v,d+1,D); } } void solve2(int s,int d){ used[s]=true; dep[s]=d; build(s,-1,0,d); int g=0; rep(i,E[s].size()){ int u=E[s][i]; if(used[u])continue; sum2[s].push_back({}); cnt2[s].push_back({}); id[G[s][g]]=g; g++; } for(int u:G[s]){ solve2(u,d+1); } } void update(int x,int y){ int X=x; int prv=-1; while(x!=-1){ int d=dist[dep[x]][X]; sum1[x]+=d*y;cnt1[x]+=y; if(prv!=-1)sum2[x][prv]+=d*y,cnt2[x][prv]+=y; prv=id[x]; x=par[x]; } } ll query(int x){ ll ans=0; int X=x; int prv=-1; while(x!=-1){ ll d=dist[dep[x]][X]; ans+=d*cnt1[x]; ans+=sum1[x]; if(prv!=-1){ ans-=sum2[x][prv]; ans-=d*cnt2[x][prv]; } prv=id[x]; x=par[x]; } return ans; } int c[200000]; int main(){ int n,K,q;cin>>n>>K>>q; rep(i,K){ scanf("%d",&c[i]),c[i]--; } rep(i,n-1){ int a,b;scanf("%d%d",&a,&b);a--;b--; E[a].push_back(b); E[b].push_back(a); } solve(0,-1); memset(used,false,sizeof(used)); solve2(find(par,par+n,-1)-par,0); rep(i,K)update(c[i],1); rep(i,q){ int ty;scanf("%d",&ty); if(ty==1){ int p,d;scanf("%d%d",&p,&d);p--;d--; update(c[p],-1); c[p]=d; update(c[p],1); } else{ int d;scanf("%d",&d);d--; printf("%lld\n",query(d)); } } }