PAT 朋友圈(改)并查集统计集合中最大元素数目和统计集合数目

某学校有N个学生,形成M个俱乐部。每个俱乐部里的学生有着一定相似的兴趣爱好,形成一个朋友圈。一个学生可以同时属于若干个不同的俱乐部。根据“我的朋友的朋友也是我的朋友”这个推论可以得出,如果A和B是朋友,且B和C是朋友,则A和C也是朋友。请编写程序计算一共有多少个朋友圈和最大朋友圈中有多少人。

输入格式:

输入的第一行包含两个正整数N(30000)和M(1000),分别代表学校的学生总数和俱乐部的个数。后面的M行每行按以下格式给出1个俱乐部的信息,其中学生从1~N编号:

第i个俱乐部的人数Mi(空格)学生1(空格)学生2 … 学生Mi

输出格式:

输出给出两个整数a,b,a表示朋友圈的数目,b表示在最大朋友圈中有多少人。

输入样例:

7 4
3 1 2 3
2 1 4
3 5 6 7
1 6

输出样例:

2 4



解析:

并查集,需要在并查集的基础上增加一个统计每个集合数目的数组。该数组在初始化的时候初始化为1,表示该集合中元素数目为1。在合并集合的时候将A集合元素的数目与B集合元素的数目相加作为合并后集合的数目。最后完成后扫一遍parent数组,如果parent[i]和i相等那么集合数目加1。

这道题是被我改的,这里算是用了并查集的基础上做了两种特别的操作,用来统计集合数目,和集合中最大的元素个数。


#include
using namespace std;

int parent[30001];
int cnt=0;		//朋友圈总数
int Num[30001];	//朋友圈人数
int MAX=0; 

void init()
{
	for(int i=0;i<1001;++i)
	{
		parent[i]=i; 
		Num[i]=1;
	}
} 

int Find(int i)
{
	if(parent[i]==i)
		return i;
	return parent[i]=Find(parent[i]);
} 

void Union(int a,int b)
{
	int A=Find(a);
	int B=Find(b);
	if(A!=B)
	{
		parent[A]=B;
		Num[B]=Num[A]+Num[B];
		if(Num[B]>=MAX)
			MAX=Num[B];
	}
	
}

int main()
{
	init();
	int n,m;
	cin>>n>>m;
	for(int i=0;i>k;
		if(k>=1)
		{
			int t=0;
			cin>>t;
			for(int l=1;l>temp;
				Union(t,temp);
			}
		}
	}
	for(int i=1;i<=n;++i)		//统计集合数目 
	{
		if(parent[i]==i)
			cnt++;
	}
	cout<


你可能感兴趣的:(编程学习,个人体会,并查集)