并查集模板

定义：
并查集是一种树型的数据结构，用于处理一些不相交集合的合并及查询问题（即所谓的并、查）。比如说，我们可以用并查集来判断一个森林中有几棵树、某个节点是否属于某棵树等。

主要构成：
并查集主要由一个整型数组pre[ ]和两个函数find( )、join( )构成。
数组 pre[ ] 记录了每个点的前驱节点是谁，函数 find(x) 用于查找指定节点 x 属于哪个集合，函数 join(x,y) 用于合并两个节点 x 和 y 。

作用：
并查集的主要作用是求连通分支数（如果一个图中所有点都存在可达关系（直接或间接相连），则此图的连通分支数为1；如果此图有两大子图各自全部可达，则此图的连通分支数为2……）

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入格式
第一行包含两个整数 N,M ,表示共有 N 个元素和 M 个操作。

接下来 M 行，每行包含三个整数 Zi​,Xi​,Yi​ 。

当 Zi​=1 时，将Xi​ 与Yi​ 所在的集合合并。

当 Zi​=2 时，输出Xi​ 与Yi​ 是否在同一集合内，是的输出 Y ；否则输出 N 。

输出格式
对于每一个Zi​=2 的操作，都有一行输出，每行包含一个大写字母，为 Y 或者 N 。

输入输出样例
输入 #1复制

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4
输出 #1复制

N
Y
N
Y
说明/提示
对于 30% 的数据，N≤10，M≤20。

对于 70% 的数据，N≤100，M≤10^3。

对于100% 的数据，1≤N≤10^4，1≤M≤2×10^5，1≤Xi​,Yi​≤N，Zi​∈{1,2}。

题目思路：

这段程序是一个使用并查集（Disjoint Set）解决元素分组问题的实现。具体来说，它通过维护一个数组 p[] 来表示每个元素所在的组，初始时每个元素自成一组。

程序主要包括两个操作：

1. 合并操作：当 z 为 1 时，将 x 和 y 所在的组合并成一个组。具体来说，程序通过调用 find() 函数找到 x 和 y 所在的组的代表元素 a 和 b，然后将 p[a] 设为 b。这样，a 所在的组就被合并到了 b 所在的组中。

2. 查询操作：当 z 为 2 时，判断 x 和 y 是否在同一个组中。具体来说，程序通过调用 find() 函数找到 x 和 y 所在的组的代表元素 a 和 b，然后判断它们是否相等。如果相等，则说明 x 和 y 在同一个组中，输出 "Y"；否则输出 "N"。

关键是理解并查集的思想和路径压缩优化。

题目代码：

#include<cstdio>
#include<iostream>
using namespace std;
int n, m;
int z, x, y;
int p[100000];
int find(int n)   
{
    return p[n] == n ? n : p[n] = find(p[n]);
}
int main()
{

    cin >> n >> m;
    for (int i = 1; i <= n; ++i)
        p[i] = i;
    for (int i = 1; i <= m; ++i) {
        cin >> z >> x >> y;
        int a = find(x), b = find(y);
        if (z == 1)
            p[a] = b;
        if (z == 2) {
            if (a == b)
                cout << "Y" << endl;
            else
                cout << "N" << endl;
        }
    }

    return 0;
}