洛谷 P3367 模板并查集

Posted 2020-09-25

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入输出格式

输入格式：

第一行包含两个整数N、M，表示共有N个元素和M个操作。

接下来M行，每行包含三个整数Zi、Xi、Yi

当Zi=1时，将Xi与Yi所在的集合合并

当Zi=2时，输出Xi与Yi是否在同一集合内，是的话输出Y；否则话输出N

输出格式：

如上，对于每一个Zi=2的操作，都有一行输出，每行包含一个大写字母，为Y或者N

输入输出样例

输入样例#1：

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

输出样例#1：

N
Y
N
Y

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据，N<=10，M<=20；

对于70%的数据，N<=100，M<=1000；

对于100%的数据，N<=10000，M<=200000。

 

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<algorithm>
 4 
 5 using namespace std;
 6 int fa[10005];
 7 int n,m;
 8 
 9 void init()
10 {
11     for(int i = 1;i <= n;i ++)
12     {
13         fa[i] = i;
14     }
15 }
16 
17 int find(int x)
18 {
19     if(fa[x] == x)
20         return x;
21     return fa[x] = find(fa[x]);
22 };
23 
24 void merge(int x,int y)
25 {
26     int a = find(x);
27     int b = find(y);
28     fa[a] = b;
29 }
30 
31 int jud(int x,int y)
32 {
33     int a = find(x);
34     int b = find(y);
35     if(a == b)
36         return 1;
37     else
38         return 0; 
39 }
40 
41 int main()
42 {
43     scanf("%d%d",&n,&m);
44     init();
45     int op,l,r;
46     for(int i = 1;i <= m;i ++)
47     {
48         scanf("%d%d%d",&op,&l,&r);
49         if(op == 1)
50         {
51             merge(l,r);
52         }
53         if(op == 2)
54         {
55             int t = jud(l,r);
56             if(t)
57                 printf("Y\n");
58             else
59                 printf("N\n");
60         }
61     }
62     return 0;
63 }