08-图7 公路村村通 (30 分)

Posted 2020-11-21 wanghao-boke

现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:

输入数据包括城镇数目正整数N（≤1000）和候选道路数目M（≤3N）；随后的M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见，城镇从1到N编号。

输出格式:

输出村村通需要的最低成本。如果输入数据不足以保证畅通，则输出−，表示需要建设更多公路。

输入样例:

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>
const int maxn = 1010;
const int INF = 100000000;

int map[maxn][maxn];
int c[maxn];
bool vis[maxn] = {0};


void init(int n);
void prim(int n);

int main()
{
    int n,m;        
    scanf("%d%d",&n,&m);
    
    init(n);
    
    int u,v;
    for (int i = 0; i < m; i++)
    {
        scanf("%d%d",&u,&v);
        scanf("%d",&map[u][v]);
        map[v][u] = map[u][v];
    }
    
    prim(n);
    
    return 0;
}

void init(int n)
{
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n; j++)
        {
            map[i][j] = INF;
        }
        map[i][i] = 0;
    }
}

void prim(int n)
{
    vis[1] = true;
    int sum = 0;
    bool flag = true;
    
    for (int i = 1; i <= n; i++)
    {
        c[i] = map[i][1];
    }
    
    for (int i = 2; i <= n; i++)
    {
        int u = -1, min = INF;
        for (int j = 1; j <= n; j++)
        {
            if (!vis[j] && c[j] < min)
            {
                u = j;
                min = c[j];    
            } 
        }
        
        if (-1 == u)
        {
            flag = false;
            break;
        }
        
        sum += min;
        vis[u] = true;
        
        for (int v = 1; v <= n; v++)
        {
            if (!vis[v] && c[v] > map[u][v])
            {
                c[v] = map[u][v];
            }
        }
    }
    
    if (flag)
    {
        printf("%d",sum);
    }
    else
    {
        printf("-1");
    }
    
    
}