結果
問題 | No.1197 モンスターショー |
ユーザー |
|
提出日時 | 2020-10-10 16:06:06 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 657 ms / 3,000 ms |
コード長 | 2,229 bytes |
コンパイル時間 | 2,000 ms |
コンパイル使用メモリ | 173,276 KB |
実行使用メモリ | 49,792 KB |
最終ジャッジ日時 | 2024-07-20 16:02:54 |
合計ジャッジ時間 | 8,998 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 41 |
ソースコード
#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)); } } }