考研复试系列——第五节 并查集

考研复试系列——第五节 并查集

前言

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时，输入结束，该用例不被处理。 

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
  
  

思考：初始时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 对朋友关系后，要求我们找出一个最大（人数最多）的集合，该集合中任意两人之间都是朋友或者该集合中只有一个人，输出该最大人数。

思考：类比上一题，我们可以看出这道题目和上一题基本一致，只是要求的结果发生了改变。上一题是要我们求集合的个数，本题是求最大的集合。

#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;
}

这里关键是对于敌人的敌人是朋友这句话的解读，别的地方与前面相似，当读入一个E a b后我们进行判断是否存在敌人的敌人这种关系，如果存在则说明存在朋友朋友的关系

于是进行合并。