首页 > 代码库 > 最小生成树(卡鲁斯卡尔)

最小生成树(卡鲁斯卡尔)

http://acm.sdut.edu.cn/sdutoj/problem.php?action=showproblem&problemid=2896

最小生成树:n个顶点n-1条边

本题因为有50000个点,所以只能用Kuscal

 

  1. #include <iostream>  
  2. #include <stdio.h>  
  3. #include <string.h>  
  4. #include <stdlib.h>  
  5. #include <math.h>  
  6. using namespace std;  
  7. int n,m,tt;  
  8. struct node  
  9. {  
  10.     int x,y,w;  
  11. }edge[200001];  
  12. int bin[50002];  
  13. void add(int u,int v,int w1)  
  14. {  
  15.     edge[tt].x=u;  
  16.     edge[tt].y=v;  
  17.     edge[tt].w=w1;  
  18.     tt++;  
  19. }  
  20. int cmp(const void *a,const void *b)  
  21. {  
  22.     return (*(struct node *)a).w-(*(struct node *)b).w;  
  23. }  
  24. int findx(int x)  
  25. {  
  26.     int r=x;  
  27.     while(r!=bin[r])  
  28.         r=bin[r];  
  29.     int j,k;  
  30.     j=x;  
  31.     while(j!=r)  
  32.     {  
  33.         k=bin[j];  
  34.         bin[j]=r;  
  35.         j=k;  
  36.     }  
  37.     return r;  
  38. }  
  39. void merge(int fx,int fy)  
  40. {  
  41.    // int fx=findx(x);  //节约时间
  42.    //int fy=findx(y);  //节约时间
  43.     if(fx!=fy)  
  44.         bin[fx]=fy;  
  45. }  
  46. void Kuscal()  
  47. {  
  48.     int sum=0;  
  49.     int ll=1;//最小生成树n-1条边  
  50.     int i=0;  
  51.     while(ll<n)  
  52.     {  
  53.         if(findx(edge[i].x)!=findx(edge[i].y))  
  54.         {  
  55.             merge(edge[i].x,edge[i].y);  
  56.             sum=sum+edge[i].w;  
  57.             ll++;  
  58.         }  
  59.         i++;  
  60.     }  
  61.     printf("%d\n",sum);  
  62.   
  63. }  
  64. int main()  
  65. {  
  66.     int u,v,w1;  
  67.     while(scanf("%d%d",&n,&m)!=EOF)  
  68.     {  
  69.         tt=0;  
  70.         for(int i=1;i<=n;i++)  
  71.             bin[i]=i;  
  72.         while(m--)  
  73.         {  
  74.             scanf("%d%d%d",&u,&v,&w1);  
  75.             add(u,v,w1);  
  76.         }  
  77.         qsort(edge,tt,sizeof(edge[0]),cmp);  
  78.        Kuscal();  
  79.     }  
  80.     return 0;  
  81. }  
  82.   
  83.   
  84.    
  85.