find函数
用于查找一个数字的祖宗数字
比如 初始时,每个数字的祖宗是自己 find(i) = i
若 要把 3 和 4 合并 则把3的祖宗设置为4 此时 p(find(3)) = 4
若 要把 5 和 3 合并, 则先找到5和3的祖宗, 再把5的祖宗设置为3的祖宗
p[find(5)]= find(3)
int find(int x) {
if (p[x] != x) p[x] = find(p[x]);
return p[x];
}
每次合并集合时只需要更改祖宗元素即可
合并集合
一共有 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。
每个结果占一行。
#include <iostream>
using namespace std;
int p[100100];
int find(int a){
if(p[a]!=a){
p[a] = find(p[a]);
}
return p[a];
}
int main(){
int n,m;
cin>>n>>m;
int i=0;
for(i=1; i<=n;i++){
p[i] = i;
}
while(m--){
string str;
int a,b;
cin>>str>>a>>b;
if(str[0]=='M'){
p[find(a)] = find(b);
}
else{
if(find(a)==find(b)){
cout<<"Yes"<<endl;
}
else{
cout<<"No"<<endl;
}
}
}
}
连通块中的点的数量
给定一个包含 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所在连通块中点的数量
每个结果占一行。
思路 在find的基础上加一个s数组,存储以find[i]为祖宗的集合的所有数字的个数
# include<iostream>
using namespace std;
int p[100010];
int s[100010];
int find(int x){
if(p[x]!=x) p[x] = find(p[x]);
return p[x];
}
int main(){
int n,m;
cin>>n>>m;
int i=0;
for(i=1; i<=n;i++){
p[i] = i;
s[i] = 1;
}
while(m--){
string str;
int a,b;
cin>>str>>a;
if(str[0]=='C'){
cin>>b;
if(find(a)==find(b)) continue;
s[find(b)]+= s[find(a)];
p[find(a)] = find(b);
}
else if(str[1]=='1'){
cin>>b;
if(find(a)!=find(b)){
cout<<"No"<<endl;
}
else{
cout<<"Yes"<<endl;
}
}
else{
cout<<s[find(a)]<<endl;
}
}
}