7-6 公路村村通 (30 分)

现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。
输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。
输入样例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例:
12

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#define MAX 1010
using namespace std;
int a[1002][1002];//村庄和村庄之间路的钱
int vis[1002];//标记
int n,m;
int findmin()
{
    int minV;
    int mindist = MAX;
    for(int V = 1; V <= n; V++)
        if(vis[V]&& vis[V] < mindist)
        {
            mindist = vis[V];
            minV = V;//和村1最便宜的村庄
        }
    if(mindist != MAX)
        return minV;
    else
        return -1;
}
int prim(int x)
{
    int V;
    for(V = 1; V <= n; V++)
        vis[V] = a[1][V];//第一个村庄到所有村庄的钱
    int s = 0,Vcount = 0;
    vis[1] = 0;
    Vcount++;
    while(1)
    {
        V = findmin();
        if(V < 0)
            break;
        s += vis[V];
        Vcount++;//联通数
        vis[V] = 0;//标记这个村庄去过
        for(int W = 1; W <= n; W++)
            if(a[V][W] < MAX)//村庄之间可连
                if(vis[W] && a[V][W] < vis[W])
                    vis[W] = a[V][W];
    }
    if(Vcount < n)
        return -1;
    return s;
}
int main(void)
{
    int v1,v2,weight;
    cin>>n>>m;
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++)
            a[i][j] = MAX;
    while(m--)
    {
        scanf("%d%d%d",&v1,&v2,&weight);
        a[v1][v2] = a[v2][v1] = weight;
    }
    int s = prim(1);
    printf("%d",s);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43589396/article/details/88616514