【模板】并查集
题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式
第一行包含两个整数 N , M N,M N,M ,表示共有 N N N 个元素和 M M M 个操作。
接下来 M M M 行,每行包含三个整数 Z i , X i , Y i Z_i,X_i,Y_i Zi,Xi,Yi 。
当 Z i = 1 Z_i=1 Zi=1 时,将 X i X_i Xi 与 Y i Y_i Yi 所在的集合合并。
当
Z
i
=
2
Z_i=2
Zi=2 时,输出
X
i
X_i
Xi 与
Y
i
Y_i
Yi 是否在同一集合内,是的输出
Y
;否则输出 N
。
输出格式
对于每一个
Z
i
=
2
Z_i=2
Zi=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 % 30\% 30% 的数据, N ≤ 10 N \le 10 N≤10, M ≤ 20 M \le 20 M≤20。
对于 70 % 70\% 70% 的数据, N ≤ 100 N \le 100 N≤100, M ≤ 1 0 3 M \le 10^3 M≤103。
对于 100 % 100\% 100% 的数据, 1 ≤ N ≤ 1 0 4 1\le N \le 10^4 1≤N≤104, 1 ≤ M ≤ 2 × 1 0 5 1\le M \le 2\times 10^5 1≤M≤2×105, 1 ≤ X i , Y i ≤ N 1 \le X_i, Y_i \le N 1≤Xi,Yi≤N, Z i ∈ { 1 , 2 } Z_i \in \{ 1, 2 \} Zi∈{1,2}。
视频解析
#include <bits/stdc++.h>
using namespace std;
const int N = 10005;
vector<int> father(N);
vector<int> siz(N, 1); // 记录以自身为根结点的集合大小,即他及他的部下数量,初始化为1,表示各自为王
int find(int x)
{ // 找x的老大
if (x == father[x])
{ // x的老大是他自己
return x;
}
else
{
father[x] = find(father[x]); // x的老大是他老大的老大,向上递归寻找最高负责人,顺便进行赋值操作,减少后续递归次数,压缩路径
return father[x];
}
}
void unionset(int x, int y)
{ // 合并x,y所在的集合
x = find(x);
y = find(y); // 找出各自的根节点
if (x == y)
{
return; // 已经在同一集合
}
if (siz[x] > siz[y])
{ // 这里默认将小集合往大集合合并
swap(x, y);
}
// y所在的集合为大集合
father[x] = y; // x指向y
siz[x] += siz[y]; // x的集合大小增加
}
void solve()
{
int n, m;
cin >> n >> m;
for (int i = 1; i <= n; i++)
{
father[i] = i; // 初始化各自为王
}
for (int i = 1; i <= m; i++)
{
int z, x, y;
cin >> z >> x >> y;
if (z == 1)
{ // 做合并操作
unionset(x, y);
}
else if (z == 2)
{
if (find(x) == find(y))
{
cout << "Y" << endl;
}
else
{
cout << "N" << endl;
}
}
}
}
int main()
{
int t;
// cin >> t;
t = 1;
while (t--)
{
solve();
}
return 0;
}