Gym 101142G : Gangsters in Central City(DFS序+LCA+set)
阿新 • • 發佈:2018-08-11
輸出 wap 關系 can printf ngs ++ cut main
題意:現在有一棵樹,1號節點是水源,葉子節點是村莊,現在有些怪獸會占領一些村莊(即只占領葉子節點),現在要割去一些邊,使得怪獸到不了水源。給出怪獸占領和離開的情況,現在要割每次回答最小的割,使得怪獸不與1號節點有聯系,而且滿足被阻隔的村莊最少。輸出最小割與組少的被誤傷的村莊。
思路:把與一號節點相鄰的點看作祖先gfa,然後它們自己作為樹的根節點,根節點保存了子樹裏葉子節點的個數。很顯然一棵樹我們要割的是這棵樹裏所有怪獸的LCA與父親邊。子數裏所有怪獸的LCA=LCA(最小DFS序的怪獸點,最大DFS序的怪獸點),用set維護有序關系即可。
#include<bits/stdc++.h> usingnamespace std; const int maxn=200010; vector<int>G[maxn]; set<int>S[maxn]; int dfn[maxn],pos[maxn],fa[maxn][20],gfa[maxn],dep[maxn]; int sz[maxn],son[maxn],cut[maxn],num[maxn],times; void dfs(int u,int f,int two) { dfn[u]=++times; pos[times]=u; dep[u]=dep[f]+1; if(dep[u]==2) two=u; if(two) gfa[u]=two; for(int i=0;i<G[u].size();i++) if(G[u][i]!=f) dfs(G[u][i],u,two),son[u]+=sz[G[u][i]]; if(!son[u]) sz[u]=1; else sz[u]=son[u]; } int LCA(int u,int v){ if(dep[u]<dep[v]) swap(u,v); for(int i=18;i>=0;i--) if(dep[fa[u][i]]>=dep[v]) u=fa[u][i];if(u==v) return u; for(int i=18;i>=0;i--) if(fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i]; return fa[u][0]; } int ans1,ans2; int tmp[maxn]; int main() { freopen("gangsters.in","r",stdin); freopen("gangsters.out","w",stdout); int N,Q,i,j; scanf("%d%d",&N,&Q); for(i=2;i<=N;i++){ scanf("%d",&fa[i][0]); G[fa[i][0]].push_back(i); } for(i=1;i<=18;i++) for(j=1;j<=N;j++) fa[j][i]=fa[fa[j][i-1]][i-1]; dfs(1,0,0); char opt[10]; int x; while(Q--){ scanf("%s%d",opt+1,&x); int t=gfa[x]; if(opt[1]==‘+‘){ if(S[t].empty()) ans1++; else ans2-=tmp[t]; S[t].insert(dfn[x]); int Lca=LCA(pos[*S[t].begin()],pos[*S[t].rbegin()]); cut[t]=Lca; num[t]++; tmp[t]=(sz[cut[t]]-num[t]); ans2+=tmp[t]; } else { ans2-=tmp[t]; tmp[t]=0; S[t].erase(dfn[x]); num[t]--; if(S[t].empty()) ans1--,cut[t]=0; else { int Lca=LCA(pos[*S[t].begin()],pos[*S[t].rbegin()]); cut[t]=Lca; tmp[t]=(sz[cut[t]]-num[t]); ans2+=tmp[t]; } } printf("%d %d\n",ans1,ans2); } return 0; }
Gym 101142G : Gangsters in Central City(DFS序+LCA+set)