837. 连通块中点的数量
给定一个包含 nn 个点(编号为 1∼n1∼n)的无向图,初始时图中没有边。
现在要进行 mm 个操作,操作共有三种:
C a b
,在点 aa 和点 bb 之间连一条边,aa 和 bb 可能相等;Q1 a b
,询问点 aa 和点 bb 是否在同一个连通块中,aa 和 bb 可能相等;Q2 a
,询问点 aa 所在连通块中点的数量;
输入格式
第一行输入整数 nn 和 mm。
接下来 mm 行,每行包含一个操作指令,指令为 C a b
,Q1 a b
或 Q2 a
中的一种。
输出格式
对于每个询问指令 Q1 a b
,如果 aa 和 bb 在同一个连通块中,则输出 Yes
,否则输出 No
。
对于每个询问指令 Q2 a
,输出一个整数表示点 aa 所在连通块中点的数量
每个结果占一行。
数据范围
1≤n,m≤1051≤n,m≤105
输入样例:
5 5
C 1 2
Q1 1 2
Q2 1
C 2 5
Q2 5
输出样例:
Yes
2
3
难度:简单 |
时/空限制:1s / 64MB |
总通过数:26813 |
总尝试数:53877 |
来源:模板题 |
算法标签 |
受校天梯赛引发,我回想起AcWing中的这道模板题,再次尝试发现在并查集中可能会产生的错误
首先并查集一般会在Find函数里面再和其它的搭配发生新的变化
知识点:1、find函数的写法(递归通过每次连接祖宗结点降低时间复杂度为 O(log* n))
2、可能合并的是在同一区间的两个数也就是题目所说的如果两个在一个区间则忽视操作
3、先合并不同区间的数量数组再合并祖宗节点(不然产生的是已经合并的一个整体块然后数量*2)
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e5+5;
int p[N];
int b[N];
int Find(int x)
{
if(p[x]!=x)p[x]=Find(p[x]);
return p[x];
}
int n,m;
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
p[i]=i;
b[i]=1;
}
for(int i=1;i<=m;i++)
{
char op[2];
scanf("%s",op);
int a,c;
if(op[0]=='C')
{
scanf("%d%d",&a,&c);
a=Find(a),c=Find(c);
if(a!=c) //防止两个在一个结点中
{
b[Find(c)]+=b[Find(a)]; //先将两个块的数量加起来
p[Find(a)]=Find(c); //再合并两个块
}
}
else{
if(op[1]=='1')
{
scanf("%d%d",&a,&c);
if(Find(a)==Find(c))
{
printf("Yes\n");
}
else{
printf("No\n");
}
}
else if(op[1]=='2')
{
scanf("%d",&a);
printf("%d\n",b[Find(a)]);
}
}
}
return 0;
}