合并集合(C++)

题目

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

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

M a b，将编号为 a和 b的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
Q a b，询问编号为 a 和 b 的两个数是否在同一个集合中；
输入格式
第一行输入整数 n和 m

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

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

数据范围

1≤n, m≤100000

// 需要注意的  这里 的 x, y 表示的是编号, 不是集合具体数值

#include <bits/stdc++.h>
using namespace std;
const int N = 1000010;
int fa[N];
// 初始化
void init(int n)
{
    for (int i = 1; i <= n; i++)
        fa[i] = i;
}

// 查询根结点
int find(int x)
{
    if (x == fa[x])
    {
        return x;
    }
    else
    {
        return fa[x] = find(fa[x]);
    }
}

// 合并 集合
void union_set(int x, int y)
{
    // find(y) 表示找到 y 对应的 根结点
    // fa[find(x)] 表示 x 根结点 的父亲结点
    fa[find(x)] = find(y);
}

int main()
{
    // 加快速度  不过使用 scanf 会更好一点
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);

    int n, m;
    cin >> n >> m;
    init(n);
    for (int i = 0; i < m; i++)
    {
        char op[5];
        int x, y;
        cin >> op >> x >> y;
        if (op[0] == 'Q')
        {
            if (find(x) == find(y))
                cout << "Yes" << endl;
            else
                cout << "No" << endl;
        } 
        else {
            union_set(x, y);
        }
    }

    return 0;
}


/*
测试用例
4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4

标准结果
Yes
No
Yes
*/