并查集,使用数组实现,find函数中使用了巧妙的递归
#include<iostream> using namespace std; const int N = 100010; int p[N]; int find(int k){ if(p[k] != k) p[k] = find(p[k]); return p[k]; } int main(){ int m,n; scanf("%d%d",&m,&n); for(int i = 1; i <= m; i++) p[i] = i; while(n --){ char t[2]; int a, b; scanf("%s%d%d",t,&a,&b); if(*t == 'M')p[find(a)] = find(b); else { if(find(a) == find(b)) puts("Yes"); else puts("No"); } } return 0; }