L - 还是畅通工程 HDU - 1233

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

Input

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

Output

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

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


        
  
Huge input, scanf is recommended.

最小生成树水题

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <cstring>
using namespace std;
const int maxn = 1010;
const int inf = 0x3f3f3f3f;
int g[maxn][maxn],d[maxn],vis[maxn],n,m;

int prim()
{
    memset(d,inf,sizeof(d));
    memset(vis,0,sizeof(vis));
    d[1] = 0;
    int sum = 0;

    while(1)
    {
        int v = -1;
        for(int u = 1;u <= n;u++)
        {
            if(!vis[u] && (v == -1 || d[v] > d[u])) v = u;
        }
        if(v == -1) break;
        vis[v] = 1;
        sum += d[v];
        for(int u = 1;u <= n;u++)
        {
            if(!vis[u]) d[u] = min(d[u],g[v][u]);
        }
    }
    return sum;
}

int main()
{
    while(scanf("%d",&n) && n)
    {
        memset(g,inf,sizeof(g));

        int m = (n - 1) * n / 2;
        for(int i = 1;i <= m;i++)
        {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            g[u][v] = g[v][u] = w;
        }
        int ans = prim();
        printf("%d\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Eric_chen_song_lin/article/details/82762677
今日推荐