HDU 1863 (图论基础prim算法)

左手的ㄟ右手 2022-06-06 22:12 133阅读 0赞

问题描述:

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本

Input

测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N   
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。

Output

对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。

Sample Input

3 3
    1 2 1
    1 3 2
    2 3 4
    1 3
    2 3 2
    0 100

Sample Output

3
    ?

题目题意:题目给我们m个点之间的几条路，问有没有最小生成树，有输出结果，没有输出?

代码如下:

#include<iostream>
    #include<cstdio>
    #include<cstring>
    #include<cmath>
    using namespace std;
    
    const int INF=0x3f3f3f3f;
    const int maxn=105;
    int cost[maxn][maxn];
    int mincost[maxn];
    bool vis[maxn];
    int V,res;//答案
    
    void prim()
    {
        for (int i=1;i<=V;i++) {
            mincost[i]=INF;
            vis[i]=false;
        }
        res=0;
        mincost[1]=0;
        while (true) {
            int v=-1;
            for (int u=1;u<=V;u++) {
                if (!vis[u]&&(v==-1||mincost[u]<mincost[v])) v=u;
            }
            if (v==-1)  break;
            vis[v]=true;
            res+=mincost[v];
            for (int u=1;u<=V;u++)
                mincost[u]=min(mincost[u],cost[v][u]);
        }
    }
    int main()
    {
        int m;
        while (scanf("%d%d",&m,&V)!=EOF) {
            if (m==0) break;
            for (int i=1;i<=V;i++)
                for (int j=1;j<=V;j++)
                    cost[i][j]=INF;
            for (int i=0;i<m;i++) {
                int a,b,c;
                scanf("%d%d%d",&a,&b,&c);
                cost[a][b]=cost[b][a]=c;
            }
            prim();
            if (res>INF)//大于INF就没有结果
               printf("?\n");
            else
                printf("%d\n",res);
        }
        return 0;
    }