12.2 并查集 ——【蓝桥幼儿园】

题目描述

蓝桥幼儿园的学生是如此的天真无邪，以至于对他们来说，朋友的朋友就是自己的朋友。

小明是蓝桥幼儿园的老师，这天他决定为学生们举办一个交友活动，活动规则如下：

小明会用红绳连接两名学生，被连中的两个学生将成为朋友。

小明想让所有学生都互相成为朋友，但是蓝桥幼儿园的学生实在太多了，他无法用肉眼判断某两个学生是否为朋友。于是他起来了作为编程大师的你，请你帮忙写程序判断某两个学生是否为朋友（默认自己和自己也是朋友）

输入描述

输出描述

对于每个op=2的输入，如果z和y是朋友，则输出一行YES，否则输出一行NO。

输入输出样例

输入：

5 5 
2 1 2
1 1 3
2 1 3
1 2 3 
2 1 2

输出：

NO
YES
YES

---

最终代码

1. c/c++

#include <bits/stdc++.h>
using namespace std;
const int maxn = 8e5+5;
int s[maxn];
void init_set(){                 //初始化
   for(int i = 1; i <= maxn; i++)
        s[i] = i;
}
int find_set(int x){
    if(x != s[x])
        s[x] = find_set(s[x]);   //路径压缩
    return s[x];
}
void merge_set(int x, int y){    //合并
    x = find_set(x);
    y = find_set(y);
    if(x != y) s[x] = s[y];
}
int main (){
    int op, n, m, x, y;
    init_set();
    cin >> n >> m;
    while(m--){
        cin >> op >> x >> y;
        if(op == 1)   merge_set(x, y);
        if(op == 2){
            if(find_set(x)==find_set(y))
                cout << "YES"<<endl;
            else cout << "NO"<<endl;
        }
    }
    return 0;
}

---

过程理解