并查集

一共有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”。

每个结果占一行。

数据范围

1n,m1051≤n,m≤105

输入样例:

4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4

输出样例:

Yes
No
Yes

##################################################
 1 #include <iostream>
 2 using namespace std;
 3 
 4 const int N = 1e5+10;
 5 int set[N];
 6 
 7 int find(int x){
 8     if(set[x] != x)set[x] = find(set[x]);
 9     return set[x];
10 }
11 
12 int main(){
13     int n, m;
14     cin >> n >> m;
15     for(int i = 1; i <= n;++i)
16         set[i] = i;
17     while(m--){
18         char op[2];
19         int a, b;
20         cin >> op >> a >> b;
21         if(op[0] == 'Q'){
22             if(find(a) == find(b))
23                 cout << "Yes" << endl;
24             else cout << "No" << endl;
25         }else{
26             set[find(a)] = find(b);
27         }
28     }
29     return 0;
30 }
View Code

 应用:

##   连通块中点的数量::

给定一个包含n个点(编号为1~n)的无向图,初始时图中没有边。

现在要进行m个操作,操作共有三种:

  1. “C a b”,在点a和点b之间连一条边,a和b可能相等;
  2. “Q1 a b”,询问点a和点b是否在同一个连通块中,a和b可能相等;
  3. “Q2 a”,询问点a所在连通块中点的数量;

输入格式

第一行输入整数n和m。

接下来m行,每行包含一个操作指令,指令为“C a b”,“Q1 a b”或“Q2 a”中的一种。

输出格式

对于每个询问指令”Q1 a b”,如果a和b在同一个连通块中,则输出“Yes”,否则输出“No”。

对于每个询问指令“Q2 a”,输出一个整数表示点a所在连通块中点的数量

每个结果占一行。

数据范围

1n,m1051≤n,m≤105

输入样例:

5 5
C 1 2
Q1 1 2
Q2 1
C 2 5
Q2 5

输出样例:

Yes
2
3

###############################################################
 1 #include <iostream>
 2 #include <string>
 3 using namespace std;
 4 
 5 const int N = 1e5+10;
 6 int set[N];
 7 int cnt[N];
 8 
 9 int find(int x){
10     if(set[x] != x) set[x] = find(set[x]);
11     return set[x];
12 }
13 
14 int main(){
15     int n,m;    
16     cin >> n >> m;
17     for(int i = 1;i <= n;++i){
18         set[i] = i;
19         cnt[i] = 1;
20     }
21     while(m --){
22         string op;
23         int a, b;
24         cin >> op;
25         if(op == "C"){
26             cin >> a >> b;
27             if(find(a) != find(b)){
28                 cnt[find(b)] += cnt[find(a)];
29                 set[find(a)] = find(b);
30             }
31         }
32         if(op == "Q1"){
33             cin >> a >> b;
34             if(find(a) == find(b))cout << "Yes" << endl;
35             else cout << "No" << endl;
36         }
37         if(op == "Q2"){
38             cin >> a;
39             cout << cnt[find(a)] << endl;    
40         }
41     }
42     return 0;
43 }
View Code
 

end

原文地址:https://www.cnblogs.com/sxq-study/p/12221414.html