异想之旅:本人原创博客完全手敲,绝对非搬运,全网不可能有重复;本人无团队,仅为技术爱好者进行分享,所有内容不牵扯广告。本人所有文章仅在CSDN、掘金和个人博客(一定是异想之旅域名)发布,除此之外全部是盗文!
题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式
第一行包含两个整数 N,M ,表示共有 N 个元素和 M 个操作。
接下来 M 行,每行包含三个整数 Zi,Xi,Yi 。
当 Zi=1 时,将 Xi 与 Yi 所在的集合合并。
当 Zi=2 时,输出 Xi 与 Yi 是否在同一集合内,是的输出 Y ;否则输出 N 。
输出格式
对于每一个 Zi=2 的操作,都有一行输出,每行包含一个大写字母,为 Y 或者 N 。
输入输出样例
In 1
4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4
Out 1
N
Y
N
Y
题目说了是并查集模板,我也就写上模板呗,AC了~
并查集的典型例题如下:
很明显,给每个人建立一个集合来存储他的亲戚时间和空间复杂度都极高,于是我们利用一个多叉树和森林的数据结构来优化存储。
具体原理和实现讲解资源已经很多了,不做重复劳动。我推荐看这两篇文章:
下面是我对本题给出的代码:
#include <bits/stdc++.h>
using namespace std;
int f[10001];
// 递归查找
int find(int a) {
if (f[a] != a) {
// 路径压缩:一棵深度大的树变成宽度大的树,减少递归
return f[a] = find(f[a]);
} else {
return a;
}
}
int hb(int a, int b) {
// 合并两集合
f[find(b)] = find(a);
}
int main() {
int n, m;
cin >> n >> m;
for (int i = 1; i <= n; i++) f[i] = i;
int z, x, y;
while (m--) {
cin >> z >> x >> y;
if (z == 1) {
hb(x, y);
} else {
if (find(x) == find(y))
cout << "Y\n";
else
cout << "N\n";
}
}
return 0;
}
亲自试了试,虽是模板题数据不是很水,不加路径压缩过不了。
find()
函数也可以不使用递归,直接循环来写(如上面推荐的第二篇文章就是循环实现的),理解起来可能会更困难,讲解写在注释里了,大家自己看:
// 循环解法
int find(int a) {
// a是要查询的节点,ans是返回的结果
int ans = a;
while (f[ans] != ans) {
// 循环寻找a的父节点,直到找到树根
ans = f[ans];
}
int i;
while (f[a] != a) {
// 将途径节点的父节点全部刷新为树根
i = a; // 暂存当前节点,以保证更改a后仍能找回当前选择的节点
a = f[a]; // 下一个要操作的节点编号
f[a] = ans; // 当前节点的父节点直接更新为树根
}
return ans;
}