2021-01-28

并查集

在一些有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;
}