并查集模板 ＋ 例题

并查集

初始化：将所有数的祖宗预置为自己

void init()
{
for(int i=1;i<=N;i++)
{
f[i]=i;
}
}

寻找祖宗函数：返回祖宗，并进行路径压缩：将该节点的所有父节点的祖宗指向一个

int find(int x)
{
	if(x!=f[x])f[x]=find(f[x]); //递归寻找祖宗
	return f[x]; //返回找到的祖宗
}

合并函数：将两个不同的集合合并，即将一个集合的祖宗指向另一个集合的祖宗

void Union(int a,int b)
{
	f[find(a)]=find(b);//将a的祖宗指向b的祖宗
}

7-4 朋友圈的数量 (15 分)

学校有N个学生，许多同学在学习中与生活中成为了朋友，据说朋友关系有自反性（自己和自己是朋友），有对称性（你和我是朋友，我和你一定也是朋友），还有传递性，即如果A和B是朋友，且B和C是朋友，则A和C也是朋友，于是就形成了许多的朋友圈。下列矩阵表示的四个人的朋友圈：i行 j列为1表示是直接的朋友关系，为0表示并非直接的朋友关系，是否是间接的朋友不清楚。输入样例中，0,1,2是一个朋友圈，3号自己是一个朋友圈。故共有2个朋友圈。请编写程序，根据给定的关系矩阵，计算学校的朋友圈数量。 注意：因为关系矩阵是对称的，故只录入了下三角的数据，也就是只描述了i号学生与其前边的学生的朋友关系，后边的关系未再复述（包括对角线上的自己：N个学生，仅输入了N*(N+1)/2个数据）。

输入样例：

4
1
1 1
1 0 1
0 0 0 1

输出样例：

2

解题代码：

#include<bits/stdc++.h>
using namespace std;

const int N = 10001;
int n;
int f[N];

int find(int x)
{
	if(x!=f[x])f[x]=find(f[x]);
	return f[x];
}
void unions(int a,int b)
{
	f[find(a)]=find(b);
}
void init()
{
	for(int i=1;i<=n;i++)
	f[i]=i;
}

int main()
{
	cin>>n;
	init();
	int x;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=i;j++)
		{
			cin>>x;
			if(i!=j)
			{
				if(x==1)
				{
					unions(i,j);
				}
			}
		}
	}
	
	set<int> cnt;
	for(int i=1;i<=n;i++)
	{
		

		cnt.insert(find(i));
	}
	cout<<cnt.size();

}

7-25 朋友圈 (25 分)

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

输入格式:

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

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

输出格式:

输出给出一个整数，表示在最大朋友圈中有多少人。

输入样例:

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

输出样例:

4

解题代码

#include<bits/stdc++.h>
using namespace std;

const int N = 100001;
int n,m;
int f[N];
int find(int x)
{
	if(x!=f[x])f[x]=find(f[x]);
	return f[x];
}

void Union(int a,int b)
{
	f[find(a)]=find(b);
}

void init()
{
	for(int i=0;i<=n;i++)
	f[i]=i;
}

int main()
{
	cin>>n>>m;
	init();
	for(int i=1;i<=m;i++)
	{
		int p;
		cin>>p;
		int x;
		if(p>=1)
		cin>>x;
		int y;
		for(int j=1;j<=p-1;j++)
		{
			cin>>y;
			Union(y,x);
		}
	}
	int cnt=0;
	int a[N]={0};
	for(int i=1;i<=n;i++)
	{
		a[find(i)]++;
		if(find(i)==i)cnt++;
	}
	int maxs = 0;
	for(int i=0;i<=n;i++)
	maxs=max(a[i],maxs);

	cout<<maxs;
}