1. 程式人生 > >4926-還是暢通工程(最小生成樹) ZCMU

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