[ 并查集 ] BZOJ3319

和bzoj4551一样，用并查集将白色边缩起来就好了。

#include<bits/stdc++.h>
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline void Read(int& x){
    char c=nc();
    for(;c<'0'||c>'9';c=nc());
    for(x=0;c>='0'&&c<='9';x=(x<<3)+(x<<1)+c-48,c=nc());
}
const int N=1000010;
const int INF=1e9;
int k,n,m,x,y;
int F[N],f[N],a[N],d[N];
int nx[N<<1],t[N<<1],h[N],id[N<<1],num;
int u[N],v[N],Ans[N];
bool b[N];
vector<int>g[N];
inline int find(int x){
    return F[x]==x?x:F[x]=find(F[x]);
}
inline void Add(int x,int y,int z){
    t[++num]=y;nx[num]=h[x];h[x]=num;id[num]=z;
}
void Dfs(int x,int y,int z){
    f[x]=y;a[x]=z;d[x]=d[y]+1;F[x]=x;
    for(int i=h[x];i;i=nx[i])
    if(t[i]!=y)Dfs(t[i],x,id[i]);
}
inline void Update(int x,int y,int z){
    x=find(x);y=find(y);
    while(x!=y){
        if(d[x]<d[y])swap(x,y);
        if(!b[x])g[z].push_back(x),F[x]=F[f[x]],b[x]=1;else x=f[x];
        x=find(x);
    }
}
int main(){
    Read(n);Read(m);
    for(int i=1;i<n;i++)Read(x),Read(y),Add(x,y,i),Add(y,x,i);
    Dfs(1,0,0);
    for(int i=1;i<=m;i++){
        Read(k);Read(u[i]);
        if(k==2)Read(v[i]),Update(u[i],v[i],i);
    }
    for(int i=1;i<=n;i++)if(b[i]||i==1)F[i]=i;else F[i]=f[i];
    for(int i=m;i;i--)
    if(!v[i])Ans[i]=a[find(u[i])];else{
        for(int j=0;j<g[i].size();j++)F[g[i][j]]=F[f[g[i][j]]];
    }
    for(int i=1;i<=m;i++)if(!v[i])printf("%d\n",Ans[i]);
    return 0;
}