加权并查集（Cube Stacking），种类并查集（Bug‘s Life）

纪念下吧，理解了几天的题目o(╥﹏╥)o

加权并查集题目链接：Cube Stacking
种类并查集题目链接：Bug’s Life

Cube Stacking代码

#include<iostream>
#define ll long long
#define mem(a) memset(a,0,sizeof(a))
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
using namespace std;

int f[30005],size[30005],r[30005];//r[x]表示x到根节点fx有多少方块 
//size[x]表示x所在的堆中有多少个方块 
int find(int x){
	if(x==f[x])return f[x];
	int t=find(f[x]);//先向下递归 
	r[x]=r[x]+r[f[x]];//再逐渐向上更新r[x]
	f[x]=t;
	return f[x];
}

int main()
{
	int t,x,y,fx,fy;
	string op;
	for(int i=1;i<=30000;i++){
		f[i]=i;
		r[i]=0;
		size[i]=1;
	}
	cin>>t;
	while(t--){
		cin>>op;
		if(op=="M"){
			cin>>x>>y;
			fx=find(x);
			fy=find(y);
			f[fx]=fy;//将fx这个堆放到fy上 
			r[fx]+=size[fy]; 
			size[fy]+=size[fx];//更新fy这个堆的方块数 
		}
		else{
			cin>>x;
			find(x); 
			cout<<r[x]<<endl;
			}
		}
	return 0;
}

Bug’s Life代码

#include<iostream>
#define ll long long
#define mem(a) memset(a,0,sizeof(a))
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
using namespace std;

int f[2005],r[2005];//r[i]表示i与根节点fi的关系，若r[i]=1。则表示i与根节点fi为异性，=0为同性 
int find(int x){
	if(x==f[x])return f[x];
	int t=find(f[x]);
	r[x]=(r[x]+r[f[x]])&1;//路径压缩＋更新r[x]即x与新的根节点的关系
	f[x]=t;
	return f[x];
}

int main(){
	int t,n,m,x,y;
	scanf("%d",&t);
	for(int i=1;i<=t;i++){
		scanf("%d %d",&n,&m);
		for(int j=1;j<=n;j++){
			f[j]=j;
			r[j]=0;//初始化，自己和自己是同性，所以为0 
		}
		int flag=0;
		for(int j=1;j<=m;j++){
			cin>>x>>y;
			if(flag==1)continue;
			int fx=find(x),fy=find(y);
			if(fx==fy){
				if(r[x]==r[y]){
					flag=1;continue;
				}
			}
		f[fx]=fy;//将fx的根节点设置为fy，怎么表示fx与fy的关系呢？可以通过fx与x的关系，x与y的关系，y与fy的关系推得 
		//x与y的关系为异性所以为1，fx与x的关系为r[x], y与fy的关系为r[y] 
		r[fx]=(r[x]+r[y]+1)%2;
		}
		if(flag)printf("Scenario #%d:\nSuspicious bugs found!\n\n",i);
		else printf("Scenario #%d:\nNo suspicious bugs found!\n\n",i);
	}
	return 0;
}