题目描述
如题,现在有一个并查集,你需要完成合并和查询操作。
输入输出格式
输入格式:第一行包含两个整数N、M,表示共有N个元素和M个操作。
接下来M行,每行包含三个整数Zi、Xi、Yi
当Zi=1时,将Xi与Yi所在的集合合并
当Zi=2时,输出Xi与Yi是否在同一集合内,是的话输出Y;否则话输出N
输出格式:如上,对于每一个Zi=2的操作,都有一行输出,每行包含一个大写字母,为Y或者N
输入输出样例
说明
时空限制:1000ms,128M
数据规模:
对于30%的数据,N<=10,M<=20;
对于70%的数据,N<=100,M<=1000;
对于100%的数据,N<=10000,M<=200000。
#include<bits/stdc++.h> using namespace std; int g[10000 + 10]; int n , m ; int find(int k ) { while(k != g[k]) k = g[k] = g[ g[k] ]; return k; } int main() { cin >> n >> m ; for(int i = 1; i <= n ; i ++) g[i] = i; for(int i = 1; i <= m ; i ++) { int a , b ,c; scanf("%d%d%d", &a, &b , &c); if (a == 1) { int q1 = find(b) , q2 = find(c); g[q1] = g[q2]; } else if(a == 2) { int q1 = find(b) , q2 = find(c); if(q1 == q2) cout << "Y" << endl; else cout << "N" << endl; } } return 0; }