前言
本文介绍的是并查集,它属于我们所学的数据结构树的一个特例,经常用于求解各结点之间的关联问题
1. 初始化结点
1 2 3 4 5
| void init() { for(int i = 1; i <= n; i++) father[i] = i; }
|
2. 查找一个结点的根结点
1 2 3 4 5 6
| int get(int x) { if(father[x] == x) return x; return father[x] = get(father[x]); }
|
3. 合并
1 2 3 4 5 6 7
| void merge(int x, int y) { x = get(x); y = get(y); if(x != y) father[y] = x; }
|
4. 例题(luogu)
4.1 题目背景
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
4.2 题目描述
规定:x和y是亲戚,y和z是亲戚,那么x和z也是亲戚。如果x,y是亲戚,那么x的亲戚都是y的亲戚,y的亲戚也都是x的亲戚。
4.3 输入格式
第一行:三个整数n,m,p,(n<=5000,m<=5000,p<=5000),分别表示有n个人,m个亲戚关系,询问p对亲戚关系。
以下m行:每行两个数Mi,Mj,1<=Mi,Mj<=N,表示Mi和Mj具有亲戚关系。
接下来p行:每行两个数Pi,Pj,询问Pi和Pj是否具有亲戚关系。
4.4 输出格式
P行,每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。
4.5 输入输出样例
1 2 3 4 5 6 7 8 9
| 6 5 3 1 2 1 5 3 4 5 2 1 3 1 4 2 3 5 6
|
4.6 solution
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43
| #include <iostream> using namespace std; int n, m, p, a, b, c, d, father[5005];
int get(int x) { if(father[x] == x) return x; return father[x] = get(father[x]); }
void merge(int x, int y) { x = get(x); y = get(y); if(x != y) father[y] = x; }
int main() { cin >> n >> m >> p; for(int i = 1; i <= n; i++) father[i] = i; for(int i = 0; i < m; i++) { cin >> a >> b; merge(a, b); } while(p--) { cin >> c >> d; c = get(c); d = get(d); if(c == d) cout << "Yes" << endl; else cout << "No" << endl; } return 0; }
|