本篇并查集的实现为最初级,目的是为了能够让入门的小伙伴了解并查集的思想,我会以hdu1213为例子,给大家详细解释。
原题为英文,为了方便小伙伴们,我提炼了一下题意。
有n个人一起吃饭,有些人相互认识。认识的人想坐在一起,不想跟陌生人坐。例如A认识B,B认识C,那么A,B,C会坐在一张桌子上。给出认识的人,问需要多少张桌子。
输入:
T<=25,代表输入数据组数
1<=N,M<=1000,N代表人数,M代表这两个人互相认识。
接下来M行,每行输入两个数字A,B,中间有空格。
输出:每组输出一个数字,加上换行(杭电Oj对格式比较严格不按要求可能会PE)
并查集核心:
查找:
找到该下标所在的集合
int find_set(int x)
{
return x==people[x]?x:find_set(people[x]);//此处为查找
}
合并:
合并两个集合
void union_set(int x,int y)
{
x = find_set(x);
y = find_set(y);
if(x!=y)people[x] = people[y];//合并两个集合
}
合并和查找复杂度均为O(n)。
完整代码:
#include<iostream>
using namespace std;
int people[1010];
int find_set(int x)
{
return x==people[x]?x:find_set(people[x]);
}
void union_set(int x,int y)
{
x = find_set(x);
y = find_set(y);
if(x!=y)people[x] = people[y];
}
int main()
{
int t;
cin>>t;
while(t--){
int n,m;
int a,b;
cin>>n>>m;
for(int i=1;i<=n;i++)pe[i] = i;//初始化
for(int i=1;i<=m;i++){
cin>>a>>b;
union_set(a,b);//合并
}
int ans = 0;
for(int i=1;i<=n;i++)
if(pe[i] == i)//统计有多少个集合
ans++;
cout<<ans<<endl;//输出答案
}
system("pause");
return 0;
}