結果
問題 | No.1197 モンスターショー |
ユーザー | autumn-eel |
提出日時 | 2020-10-10 16:00:11 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 1,698 ms / 3,000 ms |
コード長 | 3,312 bytes |
コンパイル時間 | 2,599 ms |
コンパイル使用メモリ | 183,956 KB |
実行使用メモリ | 58,888 KB |
最終ジャッジ日時 | 2023-09-27 21:31:04 |
合計ジャッジ時間 | 15,139 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 10 ms
27,804 KB |
testcase_01 | AC | 9 ms
27,960 KB |
testcase_02 | AC | 10 ms
27,876 KB |
testcase_03 | AC | 9 ms
28,000 KB |
testcase_04 | AC | 10 ms
27,872 KB |
testcase_05 | AC | 9 ms
27,912 KB |
testcase_06 | AC | 10 ms
27,976 KB |
testcase_07 | AC | 331 ms
52,416 KB |
testcase_08 | AC | 161 ms
38,492 KB |
testcase_09 | AC | 267 ms
46,772 KB |
testcase_10 | AC | 331 ms
46,332 KB |
testcase_11 | AC | 154 ms
33,872 KB |
testcase_12 | AC | 105 ms
31,084 KB |
testcase_13 | AC | 122 ms
40,008 KB |
testcase_14 | AC | 265 ms
40,440 KB |
testcase_15 | AC | 163 ms
33,552 KB |
testcase_16 | AC | 362 ms
50,268 KB |
testcase_17 | AC | 419 ms
50,564 KB |
testcase_18 | AC | 131 ms
35,092 KB |
testcase_19 | AC | 327 ms
45,900 KB |
testcase_20 | AC | 65 ms
33,820 KB |
testcase_21 | AC | 131 ms
29,604 KB |
testcase_22 | AC | 26 ms
29,548 KB |
testcase_23 | AC | 288 ms
41,476 KB |
testcase_24 | AC | 222 ms
41,232 KB |
testcase_25 | AC | 135 ms
36,664 KB |
testcase_26 | AC | 225 ms
37,300 KB |
testcase_27 | AC | 342 ms
48,068 KB |
testcase_28 | AC | 204 ms
38,632 KB |
testcase_29 | AC | 178 ms
40,972 KB |
testcase_30 | AC | 180 ms
42,968 KB |
testcase_31 | AC | 146 ms
39,440 KB |
testcase_32 | AC | 89 ms
28,656 KB |
testcase_33 | AC | 101 ms
36,676 KB |
testcase_34 | AC | 522 ms
51,888 KB |
testcase_35 | AC | 505 ms
51,952 KB |
testcase_36 | AC | 515 ms
52,036 KB |
testcase_37 | AC | 531 ms
52,016 KB |
testcase_38 | AC | 524 ms
51,900 KB |
testcase_39 | AC | 548 ms
51,880 KB |
testcase_40 | AC | 1,698 ms
58,888 KB |
testcase_41 | AC | 9 ms
27,916 KB |
testcase_42 | AC | 10 ms
27,952 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; class LCA{ public: vector<vector<pair<int,int>>>E; vector<vector<int>>par; vector<int>d1; vector<long long>d2; int MAX_LOG=22; int n; LCA(){} LCA(int N){ n=N; E=vector<vector<pair<int,int>>>(n); par=vector<vector<int>>(MAX_LOG,vector<int>(n)); d1=vector<int>(n); d2=vector<long long>(n); } void add_edge(int a,int b,int c){ E[a].push_back(make_pair(c,b)); E[b].push_back(make_pair(c,a)); } void add_edge(int a,int b){ E[a].push_back(make_pair(1,b)); E[b].push_back(make_pair(1,a)); } private: void dfs(int u,int p,int a,long long b){ par[0][u]=p;d1[u]=a;d2[u]=b; for(auto&v:E[u]){ if(v.second!=p)dfs(v.second,u,a+1,b+v.first); } } bool flag=false; void init(){ dfs(0,-1,0,0); for(int i=1;i<MAX_LOG;i++)for(int j=0;j<n;j++){ if(par[i-1][j]==-1)par[i][j]=-1; else par[i][j]=par[i-1][par[i-1][j]]; } flag=true; } public: int lca(int u,int v){ if(!flag)init(); if(d1[u]>d1[v])swap(u,v); for(int i=0;i<MAX_LOG;i++){ if((d1[v]-d1[u])>>i&1)v=par[i][v]; } if(u==v)return u; for(int i=MAX_LOG-1;i>=0;i--){ if(par[i][u]!=par[i][v]){ u=par[i][u]; v=par[i][v]; } } return par[0][u]; } long long dist(int u,int v){ int l=lca(u,v); return d2[u]+d2[v]-d2[l]*2; } }; LCA lca; vector<int>E[200000],G[200000]; bool used[200000]; int N; int centroid; int sz[200000]; int par[200000]; int id[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 solve2(int s){ used[s]=true; 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); } } void update(int x,int y){ int X=x; int prv=-1; while(x!=-1){ int d=lca.dist(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=lca.dist(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]--; } lca=LCA(n); rep(i,n-1){ int a,b;scanf("%d%d",&a,&b);a--;b--; E[a].push_back(b); E[b].push_back(a); lca.add_edge(a,b); } solve(0,-1); memset(used,false,sizeof(used)); solve2(find(par,par+n,-1)-par); 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)); } } }