BZOJ 3319 黑白樹
阿新 • • 發佈：2017-09-29
stream lan type long play hide 黑白 ostream lap 
一道神奇的並查集。
一眼看是樹剖，但是會被卡。
有人用線段樹+並查集卡過了。
然而可以做到O(n)。
抄的題解。
先把所有點往他父親的邊變黑的時間記在col中，染黑。然後把沒變黑的點用並查集合並。倒著做一遍把該時間點的邊染白，用並查集合並。
//Twenty #include<cstdio> #include<cstdlib> #include<iostream> #include<algorithm> #include<cmath> #include<cstring> #include<queue> #includeView Code<vector> typedef long long LL; const int maxn=1e6+299; using namespace std; int n,m,fir[maxn],nxt[maxn*2],to[maxn*2],id[maxn*2],p[maxn],fa[maxn],ecnt; int col[maxn],R[maxn],lz[maxn]; struct egde{ int op,x,y,ans; }e[maxn]; void add(int u,int v,int i) { nxt[++ecnt]=fir[u]; fir[u]=ecnt; to[ecnt]=v; id[ecnt]=i; nxt[++ecnt]=fir[v]; fir[v]=ecnt; to[ecnt]=u; id[ecnt]=i; } int find(int x) { return x==fa[x]?x:fa[x]=find(fa[x]); } void dfs(int x,int f) { p[x]=f; R[x]=R[f]+1; for(int i=fir[x];i;i=nxt[i]) if(to[i]!=f) { lz[to[i]]=id[i]; dfs(to[i],x); } } void change(int x,int y,int ti) {while(x!=y) { if(R[x]<R[y]) swap(x,y); if(!col[x]) fa[x]=fa[p[x]],col[x]=ti; x=fa[x]; } } void schange(int x,int y,int ti) { while(x!=y) { if(R[x]<R[y]) swap(x,y); if(col[x]==ti) fa[x]=fa[p[x]]; x=p[x]; } } int main() { //freopen(".in","r",stdin); //freopen(".out","w",stdout); scanf("%d%d",&n,&m); for(int i=1;i<n;i++) { int u,v; scanf("%d%d",&u,&v); add(u,v,i); } dfs(1,1); for(int i=1;i<=n;i++) fa[i]=i; for(int i=1;i<=m;i++) { scanf("%d%d",&e[i].op,&e[i].x); if(e[i].op==2) { scanf("%d",&e[i].y); change(e[i].x,e[i].y,i); } } for(int i=1;i<=n;i++) fa[i]=i; for(int i=1;i<=n;i++) { if(!col[i]) { fa[i]=fa[p[i]]; } } for(int i=m;i>=1;i--) { if(e[i].op==1) { int x=find(e[i].x); e[i].ans=lz[x]; } else { schange(e[i].x,e[i].y,i); } } for(int i=1;i<=m;i++) if(e[i].op==1) printf("%d\n",e[i].ans); return 0; }
BZOJ 3319 黑白樹