并查集
定义
并查集是一种树型的数据结构,用于处理一些不相交集合的合并及查询问题。常常在使用中以森林来表示。在开始时让每个元素是一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。
步骤:
初始化把每个点所在集合初始化为其自身。通常来说,这个步骤在每次使用该数据结构时只需要执行一次,无论何种实现方式,时间复杂度均为O(N)。查找查找元素所在的集合,即根节点。合并将两个元素所在的集合合并为一个集合。通常来说,合并之前,应先判断两个元素是否属于同一集合,这可用上面的“查找”操作实现。
基本函数:
initial(n):初始化有n个元素的并查集
merge(x, y):把元素 x 和元素 y 所在的集合合并,要求 x 和 y 所在的集合不相交,如果相交则不合并。
find(x):找到元素 x所在集合的根节点。
详细代码如下(C语言版):
#include <stdio.h>
int pre[1010];
int rank[1010];
int find(int x) //递归方式
{
//查找元素x所在的集合,同时压缩寻找所经过的路径
//函数运行完后的效果是所经过的结点都指向x所找到的祖先
if(x != pre[x])
pre[x] = find(pre[x]);
return pre[x];
}
/*
int find(int x) //非递归方式
{
int r; //记录根结点
int k; //暂存当前结点
int j; //暂存当前父结点
r = x;
while(r != pre[r])
r = pre[r]; //用r记录根结点
k = x;
while(k != r) //路径压缩
{
j = pre[k]; //用i暂存父节点
pre[k] = r; //当前结点指向根结点
k = j; //转至下一个结点
}
return r; //返回根节点的值
}
*/
void merge(int x, int y)
{
/*将两个集合的元素合并成一个集合*/
x = find(x); //查找x所在的集合
y = find(y); //查找y所在的集合
if(x == y)
return; //如果x和y在同一个集合中则不需要合并
/*按秩合并:将秩小的合并到大的集合中(即将矮的树合并到高的树)*/
if(rank[x] < rank[y])
pre[x] = y;
else{
pre[y] = x;
if(rank[x] == rank[y])
rank[x]++;
}
}
int main()
{
int n; //n为点的个数
int m; //m为边的个数
int i,j,k;
while(1)
{
scanf("%d", &n);
if(n == 0)
break;
scanf("%d", &m);
for(int i = 1; i <= n; i++){
/*将要用到的数组初始化*/
pre[i] = i; //初始时每个结点的祖先是自己
rank[i] = 0; //初始时树的高度为0
}
int x, y;
for(int i = 0; i < m; i++){
/*输入边,同时合并*/
scanf("%d%d", &x, &y);
merge(x, y);
}
int sum = 0;
for(int i = 1; i <= n; i++){
/*遍历计数还有多少未连通的点*/
/*若祖先是自己证明未连通(祖先除外)*/
if(pre[i] == i)
sum++;
}
printf("%d\n", sum-1);//减一是为了去掉祖先
}
return 0;
}
C++版
#include<iostream>
#include<cstring>
using namespace std;
int father[1100];
int Find(int x)
{
if(x!=father[x])
father[x]=Find(father[x]);
return father[x];
}
void Join(int x,int y)
{
int fx=Find(x),fy=Find(y);
if(fx!=fy)
father[fy]=fx;
}
int main()
{
int n,m;
while(cin>>n>>m&&n!=0)
{
memset(father,0,sizeof(father));
int i,j;
for(i=1;i<=n;i++)
father[i]=i;
int a,b;
for(i=1;i<=m;i++)
{
cin>>a>>b;
Join(a,b);
}
int num=0;
for(i=1;i<=n;i++)
if(father[i]==i)
num++;
cout<<num-1<<endl;
}
}
畅通工程
Description
某省调查城镇交通状况,得到现有城镇道路统计表,表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通(但不一定有直接的道路相连,只要互相间接通过道路可达即可)。问最少还需要建设多少条道路?
Input
测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数,分别是城镇数目N ( < 1000 )和道路数目M;随后的M行对应M条道路,每行给出一对正整数,分别是该条道路直接连通的两个城镇的编号。为简单起见,城镇从1到N编号。
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时,输入结束,该用例不被处理。
Output
对每个测试用例,在1行里输出最少还需要建设的道路数目。
Sample Input
4 2
1 3
4 3
3 3
1 2
1 3
2 3
5 2
1 2
3 5
999 0
0
Sample Output
1
0
2
998