并查集相关

十八.并查集

快速的处理：

1.将两个集合合并

2.询问两个元素是否在一个集合当中

基本原理：

用树的形式来维护所有的集合。每一个集合的编号是根节点的编号（想象一棵由一个点延伸出很多杈的数，而第一个点就是根节点）。

对于每一个节点x，都存储其父节点p[x]是谁。当想求某个点属于某个集合时，通过找父节点快速找到根节点，其编号就是集合编号。根节点的父节点设置为其自身。

问题1：如何判断是否是树根？ if(p[x]!=x)

问题2：如何求x的集合编号。while(p[x]!=x) x=p[x]

问题3：如何合并两个集合？随意选定一棵树插入到另一棵树的某个位置即可。 px是x的集合编号，py是y的集合编号。使p[x]=y即可。

问题4：如何优化寻找路径呢？路径压缩。第一次寻找x的时候，把一路上找过去经历过的每一个节点的父节点都变为根节点

1.合并集合

一共有n个数，编号1~n，最开始每个数各自在一个集合中。

先进行m个操作，操作共两种： 1.“M a b”将编号为a和b的两个数所在集合合并，若已在同一集合中则忽略此操作

2.“Q a b”询问编号为a和b的两个数是否在同一集合中,在输出“YES”，否则“NO”

#include<iostream>
​
using namespace std;
​
const int N=100010;
​
int p[N];
int n,m;
​
int find(int x){//返回x所在集合的编号，以及路径压缩
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
​
int main(){
    scanf("%d%d",&n,&m);
    
    for(int i=1;i<=n;i++) p[i]=i; //首先把所有数的父节点设为自己
    while(m--){
        char op[2];
        int a,b;
        scanf("%s%d%d",op,&a,&b);
        
        if(op[0]=='M') p[find(a)] =find(b);//合并操作
        else{
            if(find(a)==find(b)) puts("Yes");
            else puts("NO");
        }
    }
    return 0;
}

2.连通块中点的数量

连通块：如果两个点a、b，从a可以到b，从b也可以到a则为连通块。换句话说就是连了线

给定一个包含n个点（编号1~n）的无向图，初始图中没有边

现在进行m个操作，有三种操作：

1.“C a b"在点a和点b之间连一条边，a和b可能相等

2."Q1 a b"询问a和b是否在一个连通块中，a和b可能相等

3.“Q2 a”询问a所在连通块中点的数量；

#include<iostream>
​
using namespace std;
​
const int N=100010;
​
int p[N],size[n];//size记录的是每一个集合中点的数量，实际上只有根节点的size是有意义的
int n,m;
​
int find(int x){//返回x所在集合的编号，以及路径压缩
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
​
int main(){
    scanf("%d%d",&n,&m);
    
    for(int i=1;i<=n;i++) {
        size[i]=1//每个集合的数量初始为1
        p[i]=i; //首先把所有数的父节点设为自己
    }
    while(m--){
        char op[5];
        int a,b;
       
        if(op[0]=='M'){
            scanf("%s%d%d",op,&a,&b);
            if(find(a)==find(b)) continue;//如果a和b已经在同一个集合里，则不需要执行接下来的操作
            size[find(a)]+=size[find(b)];//记录树中节点数量
            p[find(a)] =find(b);//合并操作
        } 
        else if(op[1]=='1'){
            scanf("%d%d",&a,&b);
            if(find(a)==find(b)) puts("Yes");
            else puts("NO");
        }
        else{
            scanf("%d",&a);
            printf("%d\n",size[find(a)]);
        }
    }
    return 0;
}