并查集(按秩合并+非递归路径压缩)模板题 Luogu 1551 亲戚
最近做了一道题,发现 按秩合并 真的非常重要,于是根据网上的博客学了按秩合并,虽然不明白原理是什么QAQ~,但做过许多测试,发现按秩合并就是跑得快[汗]。
1 #include<cstdio> 2 #include<iostream> 3 #include<cmath> 4 #include<algorithm> 5 #include<string> 6 #include<cstring> 7 using namespace std;
8 int n,m,p; 9 int fa[200000]; 10 int depth[200000]; 11 int a,b; 12 13 int find(int x){ //路径压缩 14 int r=x; 15 while(fa[r]!=r) r=fa[r]; 16 int p=x; 17 while(p!=r){ 18 fa[p]=r; 19 p=fa[p]; 20 } 21 return r; 22 } 23 24 void meg(int x,int y){ //合并 25 int fx,fy; 26 fx=find(x); 27 fy=find(y); 28 if(fy!=fx){ 29 if(depth[fx]<=depth[fy]){ //按秩合并 30 fa[fx]=fy; 31 if(depth[fx]==depth[fy]){ 32 depth[fy]++; 33 } 34 }else{ 35 fa[fy]=fx; 36 } 37 } 38 } 39 40 int main(){ 41 scanf("%d %d %d",&n,&m,&p); 42 for(int i=1;i<=n;i++) fa[i]=i; 43 for(int i=1;i<=n;i++) depth[i]=1; 44 for(int i=1;i<=m;i++){ 45 scanf("%d %d",&a,&b); 46 meg(a,b); 47 } 48 for(int i=1;i<=n;i++) a=find(i); 49 for(int i=1;i<=p;i++){ 50 scanf("%d %d",&a,&b); 51 if(fa[a]==fa[b]) printf("Yes\n"); 52 else printf("No\n"); 53 } 54 return 0; 55 }