合并集合

合并集合

并查集

1、将两个集合合并

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

基本思想

每一个集合用树来表示，树根的编号就是整个集合的编号。每个节点存储他的父节点，p[x]表示x的父节点。

问题1:如何判断树根？if (p[x] == x)

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

问题3:如何合并两个集合：p[x]是x的集合编号，py是y的集合编号。p[x] = y

优化：路径压缩

Example

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

现在要进行m个操作，操作共有两种：

1. “M a b”，将编号为a和b的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
2. “Q a b”，询问编号为a和b的两个数是否在同一个集合中；

输入格式

第一行输入整数n和m。

接下来m行，每行包含一个操作指令，指令为“M a b”或“Q a b”中的一种。

输出格式

对于每个询问指令”Q a b”，都要输出一个结果，如果a和b在同一集合内，则输出“Yes”，否则输出“No”。

每个结果占一行。

数据范围

1≤n,m≤1051≤n,m≤105

输入样例：

4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4

输出样例：

Yes
No
Yes

#include <iostream>
using namespace std;

const int N = 1000010;

int n, m;
int p[N];//定义father数组， 存储每个元素父节点

int find(int x)//返回x的祖宗节点 + 路径压缩
{//p[x] != x 表明p[x]不是根节点，让他的父节点=祖宗节点
    if(p[x] != x) p[x] = find(p[x]);//如果p[x]不是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);//将a集合的父节点指向b，则表明b是a的祖宗，即两个集合合并了
        else
        {
            if (find(a) == find(b)) puts("Yes");//判断ab是否在同一集合只要求他们的祖宗是不是一样的就行了
            else puts("No");
        }
    }
    
    return 0;
}