集合之并查集1
1、并查集简介(摘自百度百科)
并查集,在一些有N个元素的集合应用问题中,我们通常是在开始时让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。这一类问题近几年来反复出现在信息学的国际国内赛题中。其特点是看似并不复杂,但数据量极大,若用正常的数据结构来描述的话,往往在空间上过大,计算机无法承受;即使在空间上勉强通过,运行的时间复杂度也极高,根本就不可能在比赛规定的运行时间(1~3秒)内计算出试题需要的结果,只能用并查集来描述。
并查集是一种树型的数据结构,用于处理一些不相交集合(disjoint sets)的合并及查询问题。常常在使用中以森林来表示。
2、并查集入门题目——亲戚
题目背景
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
题目描述
规定:x和y是亲戚,y和z是亲戚,那么x和z也是亲戚。如果x,y是亲戚,那么x的亲戚都是y的亲戚,y的亲戚也都是x的亲戚。
输入格式
第一行:三个整数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是否具有亲戚关系。
输出格式
P行,每行一个’Yes’或’No’。表示第i个询问的答案为“具有”或“不具有”亲戚关系。
输入输出样例
输入 #1
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
输出 #1
Yes
Yes
No
解题步骤 :
用结构体表示每个人,用rnk存储秩,pre前驱结点,给每个人建立一个集合,集合初始化为他自己,即自己是自己的亲戚。
struct Node {
int rnk, pre;
void init(int x) {
rnk = 0, pre = x;
}
} T[MAX_N];
void Build(int n) {
for (int i = 1; i <= n; i++) T[i].init(i);
}
有亲戚关系的两个人的集合将被合并,并以两个集合中秩较大的那个集合的根节点作为合并后集合的根节点,这样秩就没有变化。
void Unite(int x, int y) {
x = Root(x), y = Root(y);
if (x == y) return;
if (T[x].rnk > T[y].rnk) T[y].pre = x;
else {
T[x].pre = y;
if (T[x].rnk == T[y].rnk) T[y].rnk++;
}
}
通过查找是否为同一个根节点来判断两个人是否在一个集合中,即是否有亲戚关系。
int Root(int x) {
if (T[x].pre != x) T[x].pre = Root(T[x].pre);
return T[x].pre;
}
bool Same(int x, int y) {
return Root(x) == Root(y);
}
全部代码:
#include<bits/stdc++.h>
#define MAX_N 5005
using namespace std;
struct Node {
int rnk, pre;
void init(int x) {
rnk = 0, pre = x;
}
} T[MAX_N];
void Build(int n) {
for (int i = 1; i <= n; i++) T[i].init(i);
}
int Root(int x) {
if (T[x].pre != x) T[x].pre = Root(T[x].pre);
return T[x].pre;
}
void Unite(int x, int y) {
x = Root(x), y = Root(y);
if (x == y) return;
if (T[x].rnk > T[y].rnk) T[y].pre = x;
else {
T[x].pre = y;
if (T[x].rnk == T[y].rnk) T[y].rnk++;
}
}
bool Same(int x, int y) {
return Root(x) == Root(y);
}
int main(){
int n,m,p,x,y;
cin >> n >> m >> p;
Build(n);
while(m--){
cin >> x >> y;
Unite(x,y);
}
while(p--){
cin >> x >> y;
if(Same(x,y)) cout << "Yes" << endl;
else cout << "No" << endl;
}
}