大招秒杀
1 #include<stdio.h> 2 #include<stdlib.h> 3 typedef struct ufset *UFset; 4 struct ufset 5 { 6 int parent[100001]; 7 int root[100001]; 8 }UFS; 9 int UFfind(int e,UFset U) 10 { 11 int i,j=e; 12 while(U->root[j]==0) 13 { 14 //printf("parent[%d]=%d root=%d\n",j,U->parent[j],U->root[j]); 15 j=U->parent[j]; 16 } 17 while(j!=e) 18 { 19 i=U->parent[e]; 20 U->parent[e]=j; 21 e=i; 22 } 23 return j; 24 } 25 int UFunion(int i,int j,UFset U) 26 { 27 if(U->parent[i]<U->parent[j]) 28 { 29 U->parent[j]+=U->parent[i]; 30 U->root[i]=0; 31 U->parent[i]=j; 32 return j; 33 } 34 else 35 { 36 U->parent[i]+=U->parent[j]; 37 U->root[j]=0; 38 U->parent[j]=i; 39 return i; 40 } 41 } 42 int main() 43 { 44 UFset UF; 45 UF=(UFset)malloc(sizeof(UFS)); 46 int e,n,m,k,x,i,a,b,t1,t2; 47 scanf("%d %d %d",&n,&m,&k); 48 for(e=1;e<=n+1;e++) 49 { 50 UF->parent[e]=1; 51 UF->root[e]=1; 52 } 53 for(i=0;i<m;i++) 54 { 55 scanf("%d %d",&a,&b); 56 t1=UFfind(a,UF);t2=UFfind(b,UF); 57 if(t1!=t2) 58 { 59 UFunion(t1,t2,UF); 60 } 61 } 62 for(i=0;i<k;i++) 63 { 64 scanf("%d",&x); 65 printf("%d\n",UF->parent[UFfind(x,UF)]); 66 } 67 return 0; 68 }
并查集简单又好用
