并查集:“并”:合并;“查”:查找;“集”:集合
也就是说并查集支持两个操作:
(1)合并两个集合
(2)查找两个元素是否属于同一个集合
并查集的实现其实就是通过一个数组:int father[N]
father[a]=b代表的是a的父亲是b
基本相关操作:
#include <bits/stdc++.h>
using namespace std;
//并查集的初始化
for(int i=0;i<n;i++)
{
father[i]=i;
}
//查找根结点
int findFather(int x)
{
while(x!=father[x])
{
x=father[x];
}
return x;
}
//合并
void Union(int a,int b)//将a和b所在的集合合并
{
int Afather=findFather(a);
int Bfather=findFather(b);
if(Afather!=Bfather)
{
father[Afather]=Bfather;//改变其中一个头节点指向即可
}
}
//当元素较多,查找根结点频繁的时候,可以进行下面的路径压缩
//注:一般就是在查找根结点的同时将经过的路径结点都指向根结点
int findFather2(int x)
{
int a=x;
while(x!=father[x])//将x指向根结点
{
x=father[x];
}
while(a!=father[a])
{
int z=a;
a=father[a];
father[z]=x;
}
return x;
} //这样每次查找都会降低下一次查找的时间复杂度,直到降到O(1)
简单的题目应用:给定人数n,好朋友组数m,以及具体谁和谁是好朋友(m组)
求这些人可以分为几组
输入样例:
7 5
1 2
2 3
3 1
1 4
5 6
输出样例:
3
#include <bits/stdc++.h>
using namespace std;
int father[1010];
bool flag[1010];//用来标注每组的头节点
int findFather(int x)
{
while(x!=father[x])
{
x=father[x];
}
return x;
}
void init(int n)
{
for(int i=1;i<=n;i++)
{
father[i]=i;
flag[i]=false;
}
}
void unit(int a,int b)
{
int temp1=findFather(a);
int temp2=findFather(b);
if(temp1!=temp2)
{
father[temp1]=temp2;
}
}
int main()
{
freopen("in.txt","r",stdin);
int n,m,a,b;//n代表一共有多少个人,m代表有几对朋友,a和b为朋友
cin>>n;
cin>>m;
init(n);//初始化
for(int i=0;i<m;i++)
{
cin>>a;
cin>>b;
unit(a,b);//ab之间建立联系
}
for(int i=1;i<=n;i++)
{
flag[findFather(i)]=true;
}
int ans=0;
for(int i=1;i<=n;i++)
{
cout<<flag[i]<<" ";
ans+=flag[i];
}
cout<<endl;
cout<<ans;
}