题目描述
给定一个包含 n 个点(编号为 1∼n)的无向图,初始时图中没有边。
现在要进行 m 个操作,操作共有三种:
C a b,在点 a 和点 b 之间连一条边,a 和 b 可能相等;
Q1 a b,询问点 a 和点 b 是否在同一个连通块中,a 和 b 可能相等;
Q2 a,询问点 a 所在连通块中点的数量;
输入格式
第一行输入整数 n 和 m。
接下来 m 行,每行包含一个操作指令,指令为 C a b,Q1 a b 或 Q2 a 中的一种。
输出格式
对于每个询问指令 Q1 a b,如果 a 和 b 在同一个连通块中,则输出 Yes,否则输出 No。
对于每个询问指令 Q2 a,输出一个整数表示点 a 所在连通块中点的数量
每个结果占一行。
数据范围
1≤n,m≤105
import java.io.*;
public class Main{
static int N=100010;
static int[] p=new int[N];
static int[] size=new int[N];
static int find(x){
if(p[x]!=x) p[x]=find(p[x]);
return p[x];
}
public static void main(String[] args) throws IOException{
BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
String[] s1=br.readLine().split(" ");
int n=Integer.parseInt(s1[0]);
int m=Integer.parseInt(s1[1]);
for(int i=1;i<=n;i++){
p[i]=i;
size[i]=1;
}
while(m--!=0){
s1=br.readLine().split(" ");//可以把刚刚用过的变量拿来用
if("C".equals(s1[0])){
int x=Integer.parseInt(s1[1]);
int y=Integer.parseInt(s1[2]);
if(find(x)==find(y)) continue;
size[find(y)]+=size[find(x)];
p[find(x)]=find[y];
}
else if ("Q1".equals(s1[0])){// 判断
int x = Integer.parseInt(s1[1]);
int y = Integer.parseInt(s1[2]);
System.out.println(find(x) == find(y)?"Yes":"No");// 如果两个点对应的根节点相同则属于同一个连通图
}else {
int x = Integer.parseInt(s1[1]);
System.out.println(size[find(x)]);// 求x对应的根节点的连通点数
}
}
}
}