POJ 1703 Find them,Catch them 带权并查集

给出$n$个人，然后每次给出一对$x$和$y$关系，表示$x$和$y$不在同一个阵营。或者让你查询$x$和$y$的关系，可以是同一阵营，不同阵营或者是不确定。
并查集直接维护不易维护，这个带权多维护一个到根的距离。如果不在同一个集合里面，那么无法确定关系。否则是否同一个阵营取决了距离的奇偶性。
带权并查集在find的时候，要保存下父亲节点，然后更新完父亲节点之后。再更新当前节点的距离$dis[cur]=(dis[cur]+dis[fa])$。
另外合并的时候除了要把一个根的父亲置为另外一个根，也要更新距离，这个距离可以通过矢量法来判断，$x$与$y$距离为$1$，$x$到根是$dis[x]$，$y$到根是$dis[y]$，那么两根距离就是走了一条路径，其距离为$dis[x]+1-dis[y]$。

#include<bits/stdc++.h> 
const int N=1e5+7;
int fa[N],dis[N]; 
int find(int x) {
	if(x==fa[x]) return x;
	int t=fa[x];
	fa[x]=find(fa[x]);
	dis[x]=(dis[x]+dis[t])%2;
	return fa[x];
}
void Union(int x,int y) {
	int fx=find(x);
	int fy=find(y);
	fa[fy]=fx;
	dis[fy]=(dis[y]+1-dis[x])%2;
} 
int main() {
	int T;
	scanf("%d",&T);
	while(T--) {
		int n,m;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++) fa[i]=i;
		for(int i=1;i<=n;i++) dis[i]=0;
		while(m--) {
			char opt[3];
			int x,y;
			scanf("%s%d%d",opt,&x,&y);
			if(opt[0]=='D') {
				Union(x,y);
			}
			else if(opt[0]=='A') {
				int fx=find(x);
				int fy=find(y);
				if(fx!=fy) puts("Not sure yet.");
				else if(dis[x]^dis[y]) puts("In different gangs.");
				else puts("In the same gang.");  
			}
		}
	}
	return 0;
}