并查集
在一些有N个元素的集合问题中,我们通常是在开始让每个元素构成一个单元素的集合,然后按一定顺序将属于同一组的元素所在的集合合并,其间要反复查找一个元素在哪个集合中。然而可以知道,这样空间时间复杂度极高,无法通过题目的时限,这个时候,就可以用到并查集来解答。
并查集是一种树型的数据结构,用于处理一些不相交集合的合并及查询问题。
初始化
初始化很简单,将每个点所在集合初始化为它自己。如有n个点,就将数组fa[i]=i
void init(){
for(int i=1;i<=n;i++){
fa[i]=i;
}
}
查找
这一步,我们只需要找到根节点,即元素所在的集合。就是当fa[x]等于x时,就找到了根节点,return x。反之,继续查找。
int get(int x){
if(fa[x]==x) return x;
return get(fa[x]);
}
合并
将两个不同元素所在的集合合并为一个集合。
void merge(int x,int y){
fa[get(x)] = get(y);
}
现在,让我们来看看模板的实现
思路&例题
P1551 亲戚
题目背景
若某个家族人员过于庞大,要判断两个是否是亲戚,确实还很不容易,现在给出某个亲戚关系图,求任意给出的两个人是否具有亲戚关系。
题目描述
规定: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
这道题就可以用并查集来实现。
相当于
如图,x为y的亲戚,而y又为z的亲戚,所以z为x的亲戚。
然后,就可以搞一波了:
#include<bits/stdc++.h>
using namespace std;
const int maxn=10010;
int n,m,p;
int fa[maxn];
void init(){
for(int i=1;i<=n;i++){
fa[i]=i;
}
}
int get(int x){
if(fa[x]==x) return x;
return get(fa[x]);
}
void merge(int x,int y){
fa[get(x)] = get(y);
}
int main(){
cin >> n >> m>>p;
init();
int x,y;
for(int i=1;i<=m;i++){
cin >> x >> y;
merge(x,y);
}
for(int i=1;i<=p;i++){
cin >> x >> y;
if(get(x)==get(y))cout << "Yes"<<endl;
else cout << "No" <<endl;
}
return 0;
}