7-10 公路村村通 (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<bits/stdc++.h>
using namespace std;
#define inf 9999
int w[1010][1010];
int dir[1010];
int flag[1010] = {0};
int path[1010];
int main()
{
    int m,n,i,j;
    cin>>m>>n;

    for(i=0; i<=m; i++)
    {
        for(j=0; j<=m; j++)
            w[i][j] = inf;
    }
    for(i=1; i<=m; i++) dir[i] = inf;

    for(i=1; i<=n; i++)
    {
        int x,y,d;
        cin>>x>>y>>d;
        w[x][y] = d;
        w[y][x] = d;
    }
    int sum=0,count_=0;

    for(i=1; i<=m; i++) dir[i] = w[1][i];
    flag[1] = 1;

    for(i=0; i<m-1; i++)
    {
        int min_ = inf,mini=-1;
        for(j=1; j<=m; j++)
        {
            if(!flag[j] && dir[j] < min_)
            {
                min_ = dir[j];
                mini = j;
            }
        }
        if(mini!=-1)
        {
            path[count_++] = mini;
            sum+= min_;
            flag[mini] = 1;
        }
        else {cout<<"-1";return 0;}
        for(j=1; j<=m; j++)
        {
            if(!flag[j] && w[mini][j] < dir[j])
                dir[j] = w[mini][j];
        }
    }
    cout<<sum<<endl;
//    printf("1");
//    for(j=0;j<m-1;j++) cout<<" "<<path[j];



}
发布了28 篇原创文章 · 获赞 4 · 访问量 4810

猜你喜欢

转载自blog.csdn.net/xxw2016/article/details/105737958