2013年中南大学复试-惠民工程

2013年中南大学复试机试题解

惠民工程

题目来源: http://pipioj.online/problem.php?id=1004.
题目描述
市政府“惠民工程”的目标是在全市n个居民点间之架设煤气管道（但不一定有直接的管道相连，只要能间接通过管道可达即可）。很显然最多可架设 n(n-1)/2条管道,然而实际上要连通n个居民点只需架设n-1条管道就可以了。现请你编写程序，计算出该惠民工程需要的最低成本。
输入
测试输入包含若干测试用例。每个测试用例的第1行给出居民点数目M ( < =100 )、 评估的管道条数 N；随后的 N 行对应居民点间管道的成本，每行给出一对正整数，分别是两个居民点的编号，以及此两居民点间管道的成本（也是正整数）。为简单起见，居民点从1到M编号。
输出
对每个测试用例，在1行里输出全市管道畅通所需要的最低成本。若统计数据不足以保证畅通，则输出“?”。
样例输入
3 3
1 2 1
1 3 2
2 3 4
3 1
2 3 2
样例输出
3
?

#include<cstdio>
#include<cstring>
#include<string>
#include<iostream>
using namespace std;
const int INF=0xfffff;

int cost[200][200];
int mincost[200];
bool used[200];
int n,m;

int prim()
{
    for(int i=1;i<=n;i++)
    {
        mincost[i]=INF;
        used[i]=false;
    }
    mincost[1]=0;
    int res=0;

    while(true)
    {
        int v=-1;
        for(int u=1;u<=n;u++)
        {
            if(!used[u]&&(v==-1||mincost[u]<mincost[v])) v=u;
        }
        if(v==-1) break;
        used[v]=true;
        res+=mincost[v];
        for(int u=1;u<=n;u++)
        {
            mincost[u]=min(mincost[u],cost[v][u]);
        }
    }
    return res;
}

int main()
{
    while(scanf("%d %d",&n,&m)!=EOF){
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++) cost[i][j]=INF;
        for(int i=1;i<=n;i++) cost[i][i]=0;
        int a,b,c;
        for(int i=0;i<m;i++) {
            scanf("%d %d %d",&a,&b,&c);
            cost[a][b]=cost[b][a]=c;
        }
        int tmp;
        tmp=prim();
        if(tmp>=INF) printf("?\n");
        else printf("%d\n",tmp);
    }
    return 0;
}