考研复试系列——第五节 并查集
前言
http://blog.csdn.net/dellaserss/article/details/7724401/
首先附上另一篇博文,对并查集讲解的非常清楚 。所以这里不再重造轮子,只举一些例题加深大家的理解。
并查集主要用于表示集合信息,用来确定一个集合中含有哪些元素,判断两个元素是否在同一个集合中,求集合中元素的数量等。
例题
题目一
某省调查城镇交通状况,得到现有城镇道路统计表,表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通
(但不一定有直接的道路相连,只要互相间接通过道路可达即可)。问最少还需要建设多少条道路?
Input:
测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数,分别是城镇数目N ( < 1000 )和道路数目M;随后的M行对应M条道路,每行给出一对正整数,
分别是该条道路直接连通的两个城镇的编号。为简单起见,城镇从1到N编号。
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时,输入结束,该用例不被处理。
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时,输入结束,该用例不被处理。
output:
对每个测试用例,在1行里输出最少还需要建设的道路数目。
sample input:
sample output:4 2 1 3 4 3 3 3 1 2 1 3 2 3 5 2 1 2 3 5 999 0 0
1 0 2 998
思考:初始时N个城市,每一个城市都构成一个集合,显然此时该集合中只有一个元素。然后如果两个城市之间存在道路,则将这两个城市所属的集合进行并运算,形成一个新的集合(在每一个集合中所有的城市之间都是可达的)。最终如果有k个集合,则可知需要k-1条路才可以使这些城市相互连通。由此可以看出这就是一道经典的并查集问题。
#include<stdio.h>
#include<string>
int pre[1050];
bool t[1050];
int Find(int x)
{
int r = x;
while(r != pre[r])
r = pre[r];
int i=x,j;
while(pre[i] != r)//路径压缩
{
j = pre[i];
pre[i] = r;
i = j;
}
return r;
}
void Join(int x,int y)
{
int fx = Find(x),fy = Find(y);
if(fx != fy)
{
pre[fy] = fx;
}
}
int main()
{
int N,M,a,b,i,ans;
while(scanf("%d%d",&N,&M) && N)
{
for(i=1;i<=N;i++)
pre[i] = i;
for(i=1;i<=M;i++)
{
scanf("%d%d",&a,&b);
Join(a,b);
}
memset(t,0,sizeof(t));
for(i=1;i<=N;i++)
{
t[Find(i)] = 1;
}
for(ans=0,i=1;i<=N;i++)
{
if(t[i])
ans++;
}
printf("%d\n",ans-1);
}
return 0;
}
注:上面的代码来源于http://blog.csdn.net/dellaserss/article/details/7724401/
题目二
有 m个小朋友(m<=10000000),他们之中有M 对好朋友,且朋友关系具有传递性:若 A 与 B 是朋友,B 与 C 是朋友,那么我们也认为 A 与 C 是朋友。
在给出这 N 对朋友关系后,要求我们找出一个最大(人数最多)的集合,该集合中任意两人之间都是朋友或者该集合中只有一个人,输出该最大人数。
在给出这 N 对朋友关系后,要求我们找出一个最大(人数最多)的集合,该集合中任意两人之间都是朋友或者该集合中只有一个人,输出该最大人数。
思考:类比上一题,我们可以看出这道题目和上一题基本一致,只是要求的结果发生了改变。上一题是要我们求集合的个数,本题是求最大的集合。
#include<iostream>
using namespace std;
#define N 10000001
int pre[N];
int count[N];//用count[i]表示节点i为根的集合的元素个数
int Find(int x)
{
int r = x;
while(r != pre[r])
r = pre[r];
int i = x,j;
while(pre[i] != r)
{
j = pre[i];
pre[i] = r;
i = j;
}
return r;
}
void Join(int x,int y)
{
int fx = Find(x),fy=Find(y);
if(fx != fy)
{
pre[fy] = fx;
count[fx] += count[fy];//集合的个数保存到根上
}
}
int main()
{
int m,M;
cin>>m>>M;
int i,a,b;
for(i=1;i<=m;i++)//初始化
{
pre[i] = i;
count[i] = 1;//初始,每个集合只有一个元素,所有个数为1
}
for(i=1;i<=M;i++)
{
cin>>a>>b;
Join(a,b);
}
int ans = 0;
for(i=1;i<=m;i++)//查找最大的集合
{
if(count[i] > ans)
ans = count[i];
}
cout<<ans<<endl;
return 0;
}
题目三
来个复杂点的题目:
1920年的芝加哥,出现了一群强盗。如果两个强盗遇上了,那么他们要么是朋友,要么是敌人。而且有一点是肯定的,就是:
我朋友的朋友是我的朋友;
我敌人的敌人也是我的朋友。
两个强盗是同一团伙的条件是当且仅当他们是朋友。现在给你一些关于强盗们的信息,问你最多有多少个强盗团伙。
输入文件gangs.in的第一行是一个整数N(2<=N<=1000),表示强盗的个数(从1编号到N)。 第二行M(1<=M<=5000),表示关于强盗的信息条数。 以下M行,每行可能是F p q或是E p q(1<=p q<=N),F表示p和q是朋友,E表示p和q是敌人。输入数据保证不会产生信息的矛盾。
输出描述
Output Description
输出文件gangs.out只有一行,表示最大可能的团伙数。
样例输入
Sample Input
6
4
E 1 4
F 3 5
F 4 6
E 1 2
Sample Output
3
Data Size & Hint
2<=N<=1000
1<=M<=5000
1<=p q<=N
#include<iostream>
#include<fstream>
using namespace std;
int pre[1001];
int enemy[1001];
int ans;
int Find(int x)
{
int r = x;
while(pre[r] != r)
r = pre[r];
int i = x,j;
while(pre[i] != r)
{
j = pre[i];
pre[i] = r;
i = j;
}
return r;
}
void Fjoin(int x,int y)
{
int fx = Find(x),fy = Find(y);
if(fx != fy)
{
pre[fy] = fx;
ans--;
}
}
void Ejoin(int x,int y)
{
if(enemy[x] != 0)
Fjoin(y,enemy[x]);//敌人的敌人是朋友
else
enemy[x] = y;
}
int main()
{
ifstream file("F:\\shangji\\gang.in");
if(!file.is_open())
{
cout<<"open file error!"<<endl;
exit(1);
}
int N,M;
file>>N>>M;
ans = N;
char ch;
int a,b,i;
for(i=1;i<=M;i++)
{
file>>ch>>a>>b;
if(ch == 'F')
Fjoin(a,b);
else
{
Ejoin(a,b);
Ejoin(b,a);
}
}
file.close();
cout<<ans<<endl;
return 0;
}
于是进行合并。