并查集(C++实现初级)

本篇并查集的实现为最初级,目的是为了能够让入门的小伙伴了解并查集的思想,我会以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
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;
}

你可能感兴趣的:(数据结构,算法)