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：

 #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;
 }