并查集
并查集,在一些有N个元素的集合应用问题中,有点类似于树那样,有根节点,然后从根节点不断延伸,看分为几个根节点。反复查找一个元素在哪个集合中。并查集是一种树型的数据结构,用于处理一些不相交集合的合并及查询问题。常常在使用中以森林来表示。主要也是通过数组进行操作的。
我们用一个例题来解释这个算法
描述
有n个人,编号1-n。
现在有一个舞会,在舞会上,大家会相互介绍自己的朋友。
即: 如果a认识b,b认识c。那么在舞会上,a就会通过b认识到c。
现在,给出m个关系
每个关系描述:
a b
表示 编号为a和编号为b的人是朋友关系。
格式
输入格式
输入n和m
接下来m行,每行为a b
输出格式
最后问,会有多少个朋友圈。
样例
样例输入 Copy
5 3
1 2
2 3
4 5
样例输出 Copy
2
先分析样例,他是有五个人,那我们可以假设他们都互不认识,都有各自的朋友圈,那么一开始有五个朋友圈,且五个朋友圈都只有他们自己,那么我们可以定义一个数组pre[1010],且pre[i]=i;,
第一组数据是 1 2,说明1和2是朋友,再跟据靠左原则,则将pre[2]=1,
表示1,2的朋友圈合为一个,第二组数据2,3,则说明2,3在同一朋友圈,因为2在1的朋友圈中则3也在1的朋友圈中所有pre[3]=1;第三组数据同理得p[5]=4.最后统计有几个朋友圈。
数组变化过程如上图所示。
其中进行寻找是否属于同一朋友圈的代码是
int bin(int root)
{
if(cha[root]==root)
{
return root;
}
else
{
cha[root]=bin(cha[root]);
return cha[root];
}
}
合并朋友圈的操作是
void merge(int a,int b)
{
int root1,root2;
root1=bin(a);
root2=bin(b);
if(root1!=root2)
{
cha[root2]=root1;
}
}
其中还涉及到路径压缩算法,关于路径压缩算法。比如1的朋友圈中有2和3,我们只知道2和3是朋友,那么要知道3和1是不是朋友,还要知道2和1是不是朋友,那么就要一层层向上询问,如果在寻找的过程中,我们把这个关系直接弄出来那么关系就会很直观的表现出来。
就
就是相当于上图的变化,这样很直观就能知道1和3是不是朋友。
最后写一下完整代码
#include<stdio.h>
int cha[100010];
int bin(int root)
{
if(cha[root]==root)//寻找朋友圈
{
return root;
}
else
{
cha[root]=bin(cha[root]);//缩短路径的过程,递归过程
return cha[root];
}
}
void merge(int a,int b)//合并两个朋友圈的操作
{
int root1,root2;
root1=bin(a);
root2=bin(b);
if(root1!=root2)//靠左原则
{
cha[root2]=root1;
}
}
int main()
{
int n,m,i,total=0;
int a,b;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)
{
cha[i]=i;
}
while(m--)
{
scanf("%d%d",&a,&b);
merge(a,b);
}
for(i=1;i<=n;i++)
{
if(cha[i]==i)
{
total++;//统计朋友圈
}
}
printf("%d\n",total);
}