并查集 & 最小生成树详细讲解

并查集 & 最小生成树

并查集 Disjoint Sets

什么是并查集

    并查集,在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。这一类问题近几年来反复出现在信息学的国际国内赛题中,其特点是看似并不复杂,但数据量极大,若用正常的数据结构来描述的话,往往在空间上过大,计算机无法承受;即使在空间上勉强通过,运行的时间复杂度也极高,根本就不可能在比赛规定的运行时间(1~3秒)内计算出试题需要的结果,只能用并查集来描述。
    并查集(Disjoint Sets)是一种树型的数据结构,用于处理一些不相交集合的合并及查询问题。常常在使用中以森林来表示。(引自百度百科《并查集》)

简单来说,并查集的主要操作有:

   1- 合并两个不相交的集合
   2- 查询两个元素是否属于同一个集合

老样子,先上引例——

NKOJ 1205 亲戚

    或许你并不知道,你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。如果能得到完整的家谱,判断两个人是否亲戚应该是可行的,但如果两个人的最近公共祖先与他们相隔好几代,使得家谱十分庞大,那么检验亲戚关系实非人力所能及。在这种情况下,最好的帮手就是计算机。
  为了将问题简化,你将得到一些亲戚关系的信息,如同Marry和Tom是亲戚,Tom和Ben是亲戚,等等。从这些信息中,你可以推出Marry和Ben 是亲戚。请写一个程序,对于我们的关于亲戚关系的提问,以最快的速度给出答案。
输入格式
输入由两部分组成。
第一部分以N,M开始。N为问题涉及的人的个数(1 ≤ N ≤ 20000)。这些人的
编号为1,2,3,…,N。下面有M行(1 ≤ M ≤ 100000),每行有两个数ai, bi,表示已知ai和bi是亲戚。
第二部分以Q开始。以下Q行有Q个询问(1 ≤ Q ≤ 1 000 000),每行为ci,di,表示询问ci和di是否为亲戚。
输出格式
对于每个询问ci, di,若ci和di为亲戚,则输出yes,否则输出no。
样例输入
10  7
2  4
5  7
1  3
8  9
1  2
5  6
2  3
3
3  4
7  10
8  9
样例输出
yes
no
yes
传送门http://oi.nks.edu.cn/zh/Problem/Details?id=1205

    从题目中我们可以得到一些提示,它就是要让我们构建一个关系集合出来,再快速查找两个元素是否位于同一集合,这显然就与并查集的效用十分吻合。

合并的过程是怎样的(图示)?





并查集的工作原理




基于此算法如此高的时间复杂度,我们采用某种特殊的手段来优化它,这也便是并查集的核心内容——路径压缩

下面给出并查集的核心函数:

查询同时路径压缩
int GetFather(int v) {	//查询元素v所在集合的根节点  
	if (Father[v] == v)return v;	//v本身为根  
	else {  
		Father[v] = GetFather(Father[v]);	//只对v到根这条路径上的节点进行路径压缩  
		return Father[v];  
	}  
}  
合并两个集合
void Merge(int x, int y) {	//合并元素x和元素y所在集合  
	int fx, fy;  
	fx = GetFather(x);	//先找出x和y所在集合的根  
	fy = GetFather(y);	//两根不相同,说明x和y位于不同集合  
	if (fx != fy)Father[fx] = fy;	//将fy设为fx的父亲,合并两个集合  
}  

我们回到引例,我们现在可以很轻松地解决此题(伪代码)——

for (i = 1; i <= n; ++ i)Father[i] = i;	//初始化  
for (i = 1; i <= m; ++ i) {	//读入关系  
	cin >> x >> y;  
	Merge(x, y);  
}  
for (i = 1; i <= q; ++ i) {	//回答询问  
	cin >> x >> y;  
	if (GetFather(x) == GetFather(y))  
		cout << "Yes";  
	else cout << "No";  
}  (O(m))

#######提供几道并查集的简单练习:
NKOJ 3197 岛屿
NKOJ 1046 关押罪犯

并查集的启发式合并(有缘再补)


最小生成树 Minnimum Spanning Tree(MST)

什么是最小生成树

    一个有 n 个结点连通图的生成树是原图的极小连通子图,且包含原图中的所有 n 个结点,并且有保持图连通的最少的边。最小生成树可以用kruskal(克鲁斯卡尔)算法或prim(普里姆)算法求出。(引自百度百科《最小生成树》)

简言之,最小生成树就是在一个连通图中生成一棵树,刚好连通所有节点,所含边数(或边权总和最小)。

举个栗子,感受一下——

引例:村长的难题
    何老板是某乡村的村长,何老板打算给该村的所有人家都连上网。
    该村有n(1<=n<=1000)户人家,编号1到n。由于地形等原因,只有
m(1<=m<=50000)对人家之间可以相互牵线。在不同人家间牵线的长度不一定相同。比如在Ai与Bi之间牵线需要Ci米长的网线。
    整个村的网络入口在1号人家,何老板的问题是:是否能使得所有人家都连上网?使所有人家都连上网,最少需要多少米网线?

关注这个例子,我们的解法是要用网线连接n户人家,找出一种方案,使得总的长度最少。

我们目测可到样图的最小生成树

可是我们怎么用算法做到呢?

接下来我会介绍三种算法:
 1- Kruskal(克鲁斯卡尔算法)
 2- Prim(普里姆算法)
 3- Boruvka算法
    这三种算法都是基于贪心思想的应用,但其中Kruskal可处理同权边的情况,而Boruvka不可以。

Kruskal

Kruskal算法的基本思想:

    每次选不属于同一生成树的且权值最小的边的顶点,将边加入生成树,并将所在的2个生成树合并,直到只剩一个生成树。
    排序使用Quicksort
    检查是否在同一生成树用并查集
    总时间复杂度O(mlogm),其中m表示边的数量

以下是用Kruskal解决引例的代码:

#define maxm 10003
#define maxn 103
struct node {
	int a, b, len;	//a,b表示边的两个顶点,len表示长度  
}Edge[maxm];  //边的信息  
int n, m;	//n为顶点数,m为边数  
int Father[maxn];	//Father[]存i的父亲节点  
bool cmp(node a, node b) {	//按边长由小到大排序  
	return a.len < b.len;  
}  
void ini() {	//初始化  
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= m; ++ i)
		scanf("%d%d%d", &Edge[i].a, &Edge[i].b, &Edge[i].len);
	for (int i = 1; i <= n; ++ i)Father[i] = i;	//初始化并查集
	sort(Edge + 1, Edge + m + 1, cmp);    
}  
int GetFather(int x) {	//并查集,用于判断2个顶点是否属于同一个生成树  
	if (x != Father[x])Father[x] = GetFather(Father[x]);  
}  
void Kruskal() {
	int x, y, k, Cnt, tot;	//k为当前边的编号,tot统计最小生成树的边权总和  
	Cnt = 0;	//Cnt统计进行了几次合并,n - 1次合并就得到最小生成树  
	k = 0;
	tot = 0;
	while (Cnt < n - 1) {	//n个点构成的生成树只有n - 1条边  
		++ k;
		x = GetFather(Edge[k].a);
		y = GetFather(Edge[k].b);
		if (x != y) {
			Father[x] = y;	//合并到一个生成树  
			tot += Edge[k].len;
			++ Cnt;
		}
	}
	printf("%d", tot);
}  
int main() {
	ini();
	Kruskal();
	return 0;
}

Prim

Prim算法的基本思想:

    任选一个点,加入生成树集合。
    在未加入生成树的点中,找出离生成树距离最近的一个点,将其加入生成树。
    反复上述操作,直到所有点都加入了生成树。
    总时间复杂度O(n^2),其中n为点的个数

以下给出Prim函数代码:

void Prim(int x) {	//开始时任选一点x加入生成树,故一开始树中只有一个点x  
	int i, j, k, Min;
	int Dis[103], Path[103];
	for (i = 1; i <= n; ++ i) { 
		Dis[i] = Map[i][x];
		Path[i] = x; 
	}
	for (i = 1; i <= n - 1; ++ i) { 
		Min = inf;
		for (j = 1; j <= n; j += )
			if ((Dis[j] != 0) && (Dis[j] < Min) { 
				Min = Dis[j];
				k = j; 
		}
		Dis[k] = 0;
		for (j = 1; j <= n; ++ j)
			if( Dis[j] > Map[j][k]) { 
				Dis[j] = Map[j][k]; 
				Path[j] = k; 
			} 
	} 
}
posted @ 2019-10-19 17:35  骆孑不掉线  阅读(1409)  评论(0编辑  收藏  举报