今天学习板块-----并查集
- 合并(Union):把两个不相交的集合合并为一个集合。
- 查询(Find):查询两个元素是否在同一个集合中。
- 并查集的重要思想在于,用集合中的一个元素代表集合。我曾看过一个有趣的比喻,把集合比喻成帮派,而代表元素则是帮主。
-
最开始,所有大侠各自为战。他们各自的帮主自然就是自己。(对于只有一个元素的集合,代表元素自然是唯一的那个元素)
现在1号和3号比武,假设1号赢了(这里具体谁赢暂时不重要),那么3号就认1号作帮主(合并1号和3号所在的集合,1号为代表元素)。
-
现在2号想和3号比武(合并3号和2号所在的集合),但3号表示,别跟我打,让我帮主来收拾你(合并代表元素)。不妨设这次又是1号赢了,那么2号也认1号做帮主。
-
依次类推直到结束
几个简单代码如下
初始化
int fa[MAXN];
inline void init(int n)
{
for (int i = 1; i <= n; ++i)
fa[i] = i;
}
查询
int find(int x)
{
if(fa[x] == x)
return x;
else
return find(fa[x]);
}
合并
inline void merge(int i, int j)
{
fa[find(i)] = find(j);
}
不过这样一直进行下去的话会发现路径会拉的很长很长,时间上来看不断回溯,所以时间复杂度很高。所以需要路径压缩。。
int find(int x)
{
if(x == fa[x])
return x;
else{
fa[x] = find(fa[x]); //父节点设为根节点
return fa[x]; //返回父节点
}
}
具体题目
洛谷P3367 【模板】并查集
题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入格式
第一行包含两个整数 �,�N,M ,表示共有 �N 个元素和 �M 个操作。
接下来 �M 行,每行包含三个整数 ��,��,��Zi,Xi,Yi 。
当 ��=1Zi=1 时,将 ��Xi 与 ��Yi 所在的集合合并。
当 ��=2Zi=2 时,输出 ��Xi 与 ��Yi 是否在同一集合内,是的输出 Y
;否则输出 N
。
输出格式
对于每一个 ��=2Zi=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%30% 的数据,�≤10N≤10,�≤20M≤20。
对于 70%70% 的数据,�≤100N≤100,�≤103M≤103。
对于 100%100% 的数据,1≤�≤1041≤N≤104,1≤�≤2×1051≤M≤2×105,1≤��,��≤�1≤Xi,Yi≤N,��∈{1,2}Zi∈{1,2}。
代码实现
#include <stdio.h>
int n, m;
int pre[100000];
int find1(int x) {
while (x != pre[x]) {
x = pre[x];
}
return x;
}
int main() {
scanf("%d %d", &n, &m);
for (int i = 1; i <= n; i++) {
pre[i] = i;
}
while (m--) {
int op, x, y;
scanf("%d %d %d", &op, &x, &y);
int a, b;
a = find1(x);
b = find1(y);
if (op == 1) {
pre[a] = b;
} else if (op == 2) {
if (a == b) {
printf("Y\n");
} else {
printf("N\n");
}
}
}
return 0;
}