4926-还是畅通工程(最小生成树) ZCMU

escription

        某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

Input

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

Output

        对每个测试用例，在1行里输出最小的公路总长度。

Sample Input

8

1 2 42

1 3 68

1 4 35

1 5 1

1 6 70

1 7 25

1 8 79

2 3 59

2 4 63

2 5 65

2 6 6

2 7 46

2 8 82

3 4 28

3 5 62

3 6 92

3 7 96

3 8 43

4 5 28

4 6 37

4 7 92

4 8 5

5 6 3

5 7 54

5 8 93

6 7 83

6 8 22

7 8 17

0

Sample Output

82

思路

最小生成树模板题

代码

#include<bits/stdc++.h>
#define MAX 10005
using namespace std;
struct edge{
  int u;
  int v;
  int w;
}e[MAX];
bool cmp(edge x,edge y)
{
    return x.w<y.w;
}
int n,m,f[MAX],sum,c;
int getf(int v)
{
    if(f[v]==v)
        return v;
    else
    {
        f[v]=getf(f[v]);
        return f[v];
    }
}
int Merge(int v,int u)
{
    int t1=getf(v);
    int t2=getf(u);
    if(t1!=t2)
    {
        f[t2]=t1;
        return 1;
    }
    return 0;
}
void init()
{
    for(int i=1;i<=n;i++)
    {
        f[i]=i;
    }
}
int main()
{
    int i;
    while(~scanf("%d",&n))
    {
      if(n==0) break;
      if(n==1)
      {
         printf("0\n");
         continue;
      }
      int m=n*(n-1)/2,x;
      for(i=1;i<=m;i++)
      {
        scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
      }
      sort(e+1,e+m+1,cmp);
      init();
      c=sum=0;
      for(i=1;i<=m;i++)
      {
        if(Merge(e[i].u,e[i].v))
         {
             c+=1;
             sum+=e[i].w;
         }
         if(c==n-1)
            break;
      }
      printf("%d\n",sum);
    }
    return 0;
}