题目链接 : 点击查看
题目描述 :
一共有 n 个数,编号是 1∼n,最开始每个数各自在一个集合中。
现在要进行 m 个操作,操作共有两种:
M a b
,将编号为 a 和 b 的两个数所在的集合合并,如果两个数已经在同一个集合中,则忽略这个操作;Q a b
,询问编号为 a 和 b 的两个数是否在同一个集合中;
输入输出格式 :
输入
第一行输入整数 n 和 m。
接下来 m 行,每行包含一个操作指令,指令为
M a b
或Q a b
中的一种。
输出
对于每个询问指令
Q a b
,都要输出一个结果,如果 a 和 b 在同一集合内,则输出Yes
,否则输出No
。每个结果占一行。
输入输出样例 :
输入
4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4
输出
Yes
No
Yes
题目分析 :
并查集是一种树形结构,常用来处理一些不相交集合的分组与查询问题,其通常支持两种操作,1 . 合并 : 把两个不相交集合合并为1个集合 2. 查询 : 查询两个元素是否在同一个集合中。并查集的每一个集合都是一个树形结构,其根节点元素的下标即为整个集合的标号(下标)。要想建立并查集,我们需要一个辅助数组p[N[来记录每个节点的父节点是多少,特别地,根元素的父节点就是他本身。在初始化并查集时,我们把每个元素都当成一个单独的集合,也就是每个数都是由其自己构成集合的根节点,即初始化过程为p[i] = i。建立完毕后,我们要想查询a,b两个数是否在同一集合,就要判断他们相应集合的根节点是否为同一个,因此要 不断进行 a = p[a] , b = p[a],顺着父节点不断向上爬,直到遇到p[a] = a, p[b] = b, 如此整个代码的时间复杂度还是相当高的。这里我们可以采取路径压缩算法优化,即我们让集合内每个节点的父节点直接为根元素,这样便可在近似O(1)的时间复杂度中查询两个数是否为同一个集合,这个可以在find函数中实现,if(p[x] != x) (还没遍历到根节点) p[x] = find(p[x]) (这里是递归find函数使其不断向上移动,直到找到根节点,之后开始回溯,使得每个节点的父元素p[x]就等于(指向)根节点),find()函数返回值便是当前元素所在的集合编号(下标)。关于两个集合合并问题,我们直接将其中一个集合的根节点接到另外一个集合的根节点,即如px的根节点值为x,py根节点值为py,则令p[px] = y, 在此配合find函数的递归过程,便可令x集合中所有元素所在的集合编号为y,即p[find(a)] = find(b),相当于一棵树作为了另一棵树的分支。详见如下代码.
代码 :
#include<iostream>
#include<cstdio>
using namespace std;
const int N = 1e5 + 7;
int p[N];//储存节点n的父节点
int find(int x) {
if (p[x] != x) p[x] = find(p[x]);//路径压缩
return p[x];
}
int main() {
int n, m;
cin >> n >> m;
for (int i = 1; i <= n; i ++ ) p[i] = i;
while (m -- ) {
char op;
int a, b;
cin >> op >> a >> b;
if (op == 'M') p[find(a)] = find(b);
else {
if (find(a) == find(b)) cout << "Yes" << endl;
else cout << "No" << endl;
}
}
return 0;
}
---------------------------------------------------
在此我们给出朴素并查集的相关模板
朴素并查集:
int p[N]; //存储每个点的祖宗节点
// 返回x的祖宗节点
int find(int x)
{
if (p[x] != x) p[x] = find(p[x]);
return p[x];
}
// 初始化,假定节点编号是1~n
for (int i = 1; i <= n; i ++ ) p[i] = i;
// 合并a和b所在的两个集合:
p[find(a)] = find(b);