并查集—模板题

题目：ACWing 836. 合并集合https://www.acwing.com/problem/content/838/

一共有n个数，编号是1~n，最开始每个数各自在一个集合中。

现在要进行m个操作，操作共有两种：

  1. “M a b”，将编号为a和b的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
  2. “Q a b”，询问编号为a和b的两个数是否在同一个集合中；

输入格式

第一行输入整数n和m。

接下来m行，每行包含一个操作指令，指令为“M a b”或“Q a b”中的一种。

输出格式

对于每个询问指令”Q a b”，都要输出一个结果，如果a和b在同一集合内，则输出“Yes”，否则输出“No”。

每个结果占一行。

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<algorithm>
 4 using namespace std;
 5 const int N=100005;
 6 int fa[N];
 7 
 8 int find(int x){
 9     if(fa[x]!=x) fa[x]=find(fa[x]);
10     return fa[x];
11 }
12 
13 int main()
14 {
15     char op;
16     int a,b;
17     int n,m;
18 //    cin>>n>>m;
19     scanf("%d%d",&n,&m);
20     getchar();
21     for(int i=1;i<=n;i++) fa[i]=i;
22     while(m--){
23         scanf("%c %d%d",&op,&a,&b);
24         getchar();
25 //        cin>>op>>a>>b;
26         if(op=='M'){
27             fa[find(a)]=find(b);
28         }
29         else{
30             if(find(a)==find(b)) cout<<"Yes"<<endl;
31             else cout<<"No"<<endl;
32         }
33     }
34     return 0;
35 }

注意find函数的含义和书写；

posted @ 2020-11-10 14:19  ideologue  阅读(46)  评论(0)  编辑  收藏  举报