HLG 1133 MST【kruskal 算法】

Description

既然大家都不愿意做水题，元帅也很无奈，想了好久也不知道什么是水题，因为在元帅的眼里都是水题，他定义的非水题就是他做不出来的题。太囧了，随便弄个水的MST。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出顶点数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个顶点的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。 当N为0时，输入结束，该用例不被处理。

Output

对于每个测试用例，输出MST的最小路径总长度。

Sample Input

3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0

Sample Output

3 5

code：

View Code

 #include<stdio.h>
 #include<stdlib.h>
int r[5000],u[5000],v[5000],w[5000],p[102];
int n;
int cmp(const void*p1,const void*p2)
 {
return w[*(int*)p1]>w[*(int*)p2];        //间接排序，按边的大小边序号依次存入数组 r[i]  中
 }
int find(int x)                                     //并查集的找祖先函数
 {
int r=x;
while(p[r]!=r)
      r=p[r];
int i=x;
int j;
while(i!=r)                        //压缩路径
      {
        j=p[i];
        p[i]=r;
        i=j;
      }
return r;
 }
void join(int x,int y)              //并查集的合并函数
 {
int fx=find(x);
int fy=find(y);
if(fx!=fy)
  {
if(fx<fy)                                               
    p[fy]=fx;
else
    p[fx]=fy;
  }
 }
int kruskal()
 {
int ans=0;
int x,y,i,e;
for(i=0;i<n;i++)                         //初始化并查集
   p[i]=i;
for(i=0;i<n*(n-1)/2;i++)                 //初始化边序号 
   r[i]=i;
  qsort(r,n*(n-1)/2,sizeof(r[0]),cmp); //给边排序
  for(i=0;i<n*(n-1)/2;i++)
  {
       e=r[i];                                             
       x=find(u[e]);                           //找出当前边两个端点所在集合编号
       y=find(v[e]);
if(x!=y)                                      //如果在不同集合，合并
   {
        ans+=w[e];                                 
        join(x,y);
   }
  }
return ans;
 }
int main()
 {
int i;
while(scanf("%d",&n),n)
  {
for(i=0;i<n*(n-1)/2;i++)
    scanf("%d%d%d",&u[i],&v[i],&w[i]);
   printf("%d\n",kruskal());
  }
return 0;
 }

 

转载于:https://www.cnblogs.com/dream-wind/archive/2012/03/15/2399488.html