洛谷 P3367 【模板】并查集

好久一段时间没碰 OI 了，练个手吧

那么，今天我们就来搞一道并查集的题目吧 ！

首先是 题目连接，当然，题面我也给复制过来了（不得不说洛谷的复制 Markdown 功能确实好用）

---

题目描述

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

输入格式

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

接下来 $M$ 行，每行包含三个整数 $Z_i,X_i,Y_i$ 。

当 $Z_i=1$ 时，将 $X_i$ 与 $Y_i$ 所在的集合合并。

当 $Z_i=2$ 时，输出 $X_i$ 与 $Y_i$ 是否在同一集合内，是的输出
Y ；否则输出 N 。

输出格式

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

样例 #1

样例输入 #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\le 10$，$M\le 20$。

对于 $70\%$ 的数据，$N\le 100$，$M\le 10^3$。

对于 $100\%$ 的数据，$1\le N\le 10^4$，$1\le M\le 2\times 10^5$，$1\le X_i,Y_i\le N$， Z i ∈ { 1 , 2 } Z_i \in \{ 1, 2 \} Zi​∈{1,2}。

---

这道题嘛，要实现 $2$ 个操作，一个是查找（Find)，还有一个是合并（Union) 。

$Find$ 操作如下

int Find(int i) {
	if(root[i]==i) return root[i];
	else return root[i]=Find(root[i]); //路径压缩
}

$Union$ 操作如下

void Union(long long a,long long b) {
	// root[find(a)]=find(b);
    // 按秩合并
    if(root[find(a)]==root[find(b)]) {
        root[find(b)]==find(a);
        h[find(a)]++;
    } else {
        if(h[find(a)]<h[find(b)]) root[find(a)]=find(b);
        else root[find(b)]=find(a);
    }
}

那么其他零碎的东西就不细讲了。

AC 代码：

#include <iostream>
#include <cstdio>

using namespace std;

long long n,m;
long long root[10001],h[10001];
long long z,x,y;

long long find(long long i) {
    if(root[i]==i) return root[i];
    else return root[i]=find(root[i]);
}

void Union(long long a,long long b) {
    // 按秩合并
    if(root[find(a)]==root[find(b)]) {
        root[find(b)]==find(a);
        h[find(a)]++;
    } else {
        if(h[find(a)]<h[find(b)]) root[find(a)]=find(b);
        else root[find(b)]=find(a);
    }
    // root[find(a)]=find(b);
}

int main() {
    cin>>n>>m;
    for(long long i=1;i<=n;i++) root[i]=i,h[i]=1;
    for(long long i=1;i<=m;i++) {
        cin>>z>>x>>y;
        if(z==1) {
            Union(x,y);
        } else {
            if(find(x)==find(y)) printf("Y\n");
            else printf("N\n");
        }
    }
    return 0;
}