PAT L2-007 家庭房产（25 分）（并查集）

在学习了并查集以后拿这道题来练手非常好，涉及了并查集以及结构体二级排序算法，非常基础但是要求代码

（一开始被自己坑了，在并查集找根时应该使用递归思想）

并查集的基本操作：

//非压缩路径法找根 
int get_root(int a)
{
	if(par[a]==a)
	{
		return a;
	}
	else
	{
		return get_root(par[a]);//注意这里的递归
	}
}

//压缩路径法找根
int get_root(int a)
{
	if(par[a]!=a)
	{
		par[a] = get_root(par[a]);	
	}
	return par[a]; 
} 

void merge(int a,int b)
{
	int p1 = get_root(a);//这里要求找到根 
	int p2 = get_root(b);
	if(p1==p2)
	{
		return ;	
	} 
	else//把p1作为p2的父节点 例如这个题要求输出家庭成员序号较小的 
	{
		par[p2] = p1;
		total[p1] += total[p2];//这一步是把b的人数加到a上	
	}
}

其余操作应该可以在这两种操作基础上实现

题目链接

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int par[10005];
int total[10005] = {1};

int taoshu[10005] = {0};
int mianji[10005] ={0};

int vis[10005] = {0};

struct ren
{
	int number;
	int renshu;
	double renjutaoshu;
	double renjunmianji;
}group[10005];
int cmp(ren a,ren b)
{
	if(a.renjunmianji==b.renjunmianji)
	{
		return a.number<b.number;
	}
	else
		return a.renjunmianji>b.renjunmianji;
}


int get_root(int a)
{
	if(par[a]==a)
	{
		return a;
	}
	else
	{
		return get_root(par[a]);
	}
}
void merge(int a,int b)
{
	int p1 = get_root(a);
	int p2 = get_root(b); 
	if(p1==p2)
	{
		return ;
	}
	else
	{
		taoshu[p1] += taoshu[p2];
		mianji[p1] += mianji[p2];
		total[p1] += total[p2];
		 par[p2] = p1;
	} 
}

int main(void)
{
	int n;
	scanf("%d",&n);
	for(int i=0;i<10005;i++)//初始化 
	{
		par[i] = i;
		total[i] = 1;//每个组至少一个人 
	}
	for(int i=0;i<n;i++)
	{
		int minx;
		int b[1005];
//		printf("get_root(6666) = %d\n",get_root(6666));
		scanf("%d%d%d",&b[0],&b[1],&b[2]);
		minx =get_root(b[0]);
//		minx = b[0];
//		printf("第一次minx = %d\n",minx);
		int bb = b[1];
		int cc = b[2];
		if(b[1]!=-1&&get_root(b[1])!=-1)
		{
			bb = get_root(b[1]);
//			minx = min(minx,bb);
			if(bb<minx)
			{
//				taoshu[minx] = taoshu[bb];
//				mianji[minx] = mianji[bb];
				minx = bb;
			}
		}
		if(b[2]!=-1&&get_root(b[2])!=-1) 
		{
			cc  = get_root(b[2]); 
			if(cc<minx)
			{
//				taoshu[minx] = taoshu[cc];
//				mianji[minx] = mianji[cc];
				minx = cc;
			}
		}
		for(int j=0;j<3;j++)
		{
			if(b[j]<minx&&b[j]!=-1)
			{
//				taoshu[minx] = taoshu[get_root(b[j])];
//				mianji[minx] = mianji[get_root(b[j])];
				minx = get_root(b[j]);
			}
		}
//		printf("此时的minx = %d\n",minx);
//		system("pause");
		
		int k;
		scanf("%d",&k);
		int a[55];
		for(int j=0;j<k;j++)
		{
			scanf("%d",&a[j]);
			if(get_root(a[j])<minx)
			{
//				taoshu[minx] = taoshu[get_root(a[j])];
//				mianji[minx] = mianji[get_root(a[j])];
				minx = get_root(a[j]);
			}
		} 
		for(int j=0;j<3;j++)
		{
			if(b[j]!=-1&&get_root(b[j])!=-1)
			{
				merge(minx,get_root(b[j]));
			}
		}
		for(int j=0;j<k;j++)
		{
			merge(minx,get_root(a[j]));
		}
//		printf("5551的par = %d\n",get_root(minx));
//		system("pause");
		//现在把所有东西都加到了a的头上  这里由于要求输出家庭最小成员编号 所以要merge到最小的头上 重新输入 
		int x,y; 
		scanf("%d%d",&x,&y);
		taoshu[get_root(minx)] += x;
		mianji[get_root(minx)] += y;
//		for(int i=0;i<10005;i++)
//		{
//			if(get_root(i)==minx&&vis[i]==0)
//			{
//				total[minx] ++;
//				vis[i] = 1;
//			}
//		}	
//		printf("中间面积 = %d\n",mianji[get_root(minx)]);
//		printf("中间最小值 = %d\n",minx);
//		system("pause");
	}	
	printf("taoshu  = %d\n",taoshu[get_root(7777)]);
	printf("总人数 = %d\n",total[get_root(7777)]);
	printf("总面积 = %d\n",mianji[get_root(7777)]);
	printf("最小成员编号  = %d\n",get_root(7777)); 
	int count = 0;
	for(int i=0;i<10005;i++)
	{
		if(vis[get_root(i)]==1)
		{
			continue;
		}
		if(get_root(i)==i&&taoshu[i]==0)
		{
			continue;
		}
		vis[get_root(i)] = 1;
//			printf("\n--------------------------------------------\n");
//	printf("总套数  = %d\n",taoshu[get_root(i)]);
//	printf("总人数 = %d\n",total[get_root(i)]);
//	printf("总面积 = %d\n",mianji[get_root(i)]);
//	printf("最小成员编号  = %d\n",get_root(i)); 
//		printf("\n--------------------------------------------\n");
		group[count].number = get_root(i);
		group[count].renshu = total[get_root(i)];
		group[count].renjutaoshu = taoshu[get_root(i)]*1.0/total[get_root(i)];
		group[count].renjunmianji = mianji[get_root(i)]*1.0/total[get_root(i)];
		count++;
			
	}
	printf("%d\n",count);
	sort(group,group+count,cmp);
	for(int i=0;i<count;i++)
	{
		printf("%04d %d %.3lf %.3lf\n",group[i].number,group[i].renshu,group[i].renjutaoshu,group[i].renjunmianji);
	}
	
} 
/*
2
7777 6666 -1 0 2 300
6666 5551 5552 1 7777 1 100

*/
/*
2
2468 0001 0004 1 2222 1 500
2222 1236 2468 2 6661 6662 1 300
*/