模板/经典题型并查集维护生成树

Posted 2021-02-04 creed-qwq

这里的并查集是按秩合并并查集。
这种方法维护生成树的时候可以维护一个点到根的权值。
但是由于合并的时候做了一个类似换根一样的操作，因此这个权值通常只能是异或之类的。

//find-union-set
int top,f[N],v[N],sz[N],sk[M];
int find(int x){return x==f[x]?f[x]:find(f[x]);}
int value(int x){return x==f[x]?0:v[x]^value(f[x]);}
bool merge(int a,int b)
{
    int x=find(a),y=find(b);
    if(x==y)return false;
    if(sz[x]>sz[y])swap(x,y),swap(a,b);
    v[x]=value(a)^value(b)^1;
    f[x]=y;sz[y]+=sz[x];sk[++top]=x;
    return true;
}
void erase()
{
    int x=sk[top];
    sz[f[x]]-=sz[x];f[x]=x;v[x]=0;top--;
}