连通块:简单来说,就是从点a可以走到点b,也可以从点b走到点a,就说两个点在一个连通块中
思路:用一个集合来维护两个连通块,一个连通块中的点就在一个集合中。当在两个集合之间连一条边时,其作用就是把两个集合合并。其原理与上一篇相似
题目:
代码:
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int p[N],s[N]; //s[]表示每一个集合里点的数量,注意不要用size,避免关键字
int find(int x) {
if (x != p[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,s[i] = 1; //初始时,每个集合里只有一个点
while (m--) {
string op;
cin >> op;
int a, b;
if (op[0] == 'C'){
cin>>a>>b;
if(find(a)==find(b))continue; //如果a和b已经在一个集合里,后面操作不用进行
s[find(b)]+=s[find(a)]; //每个集合其实是一棵树,在两个集合之间连一条边时,其作用就是把两个集合合并,合并时只需在新的根节点上加上当前这棵树的点的数量
p[find(a)] = find(b);
}
else if (op[1] == '1')
{
cin>>a>>b;
if (find(a) == find(b)) puts("Yes");
else puts("No");
}
else{
cin>>a;
cout<<s[find(a)]<<endl;
}
}
return 0;
}