【模板】并查集维护生成树

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

//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--;
}
posted @ 2019-01-20 12:06  Creed-qwq  阅读(162)  评论(0编辑  收藏  举报