首页 > 代码库 > 并查集（初步）

并查集（初步）

2024-09-16 13:05:22   218人阅读

并查集可以查询某个是否在一个集合里，而且可以把两个或以上的集合合并起来。

比如有三个集合{1, 3, 5, 7, 9}, {2, 4, 6, 8, 10}, {11, 12, 13, 14},其中1和5在一个集合里2和11不在一个集合里。

经典的题目：n个人，有m对关系，如果a和b是亲戚，b和c是亲戚，则a和c也是亲戚，求有多少组亲戚。描述的不是很好，看例子吧。假设有5个人，1和2是亲戚，2和3是亲戚，4和5是亲戚。那么答案是2，因为亲戚组有两组，分别是{1, 2, 3}，{4, 5}。

树关系如下技术分享

 

在查询两个人是否是亲戚时，最要查询他们最终的根是否相同即可，比如查询1和3是否是亲戚，因为1的最终根是1,3的最终根也是1，所以1和3是亲戚。

当然，层数很多时速度很慢，那我们就可以把路径压缩下了。

技术分享

这样从3查询最终的根只要查询一次，之前要查询两次，但层数很多时，用路径压缩的速度可快了很多。

下面写下主要的代码：

 1 int find(int x){
 2     if(fa[x] == x)return x;
 3     else return fa[x] = find(fa[x]);
 4 }
 5 void uni(int x, int y){
 6     x = find(x);
 7     y = find(y);
 8     if(x == y)return;
 9     if(x < y){
10         fa[x] = y;
11     }else{
12         fa[y] = x;
13     }
14 }

其中fa[i] 表示i的父亲，第三行实现了路径压缩。

下面给出一个标准的并查集题目:

畅通工程

某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要互相间接通过道路可达即可）。问最少还需要建设多少条道路？ 

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


        
 
Huge input, scanf is recommended.

题意是求把所有的道路直接或间接的连接，最少需要修多少条道路。
代码如下：
 1 #include <iostream>
 2 #include <cstdio>
 3 using namespace std;
 4 const int maxn = 1e3+10;
 5 int fa[maxn];
 6 int n, m;
 7 int find(int x){
 8     if(fa[x] == x)return x;
 9     else return fa[x] = find(fa[x]);
10 }
11 void uni(int x, int y){
12     x = find(x);
13     y = find(y);
14     if(x == y)return;
15     if(x < y){
16         fa[x] = y;
17     }else{
18         fa[y] = x;
19     }
20 }
21 int main(){
22     int a, b;
23     while(~scanf("%d",&n)){
24         if(n == 0)break;
25         scanf("%d",&m);
26         int ans = 0;
27         for(int i = 1; i <= n; i ++) {
28             fa[i] = i;
29         }
30         for(int i = 1; i <= m; i ++){
31             scanf("%d%d",&a,&b);
32             uni(a,b);
33         }
34         for(int i = 1; i <= n; i ++){
35             if(fa[i] == i) ans++;
36         }
37         cout << ans - 1 << endl;
38     }
39     return 0;
40 }

 

并查集（初步）


联系
我们