08-图7 公路村村通

08-图7 公路村村通（30 分）

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

输入格式:

输入数据包括城镇数目正整数$N$（$\le 1000$）和候选道路数目$M$（$\le 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 <set>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>

#define mod 1000000007
const int INF = 0x3f3f3f3f;
typedef long long ll;
const int maxn = 1050;
using namespace std;

int mp[maxn][maxn];
bool vis[maxn];
int dist[maxn];
int n, m;


int findmin()
{
    int v = -1, minlen = INF;
    for(int i = 1; i <= n; i++)
    {
        if(!vis[i] && dist[i] < minlen)
        {
            v = i;
            minlen = dist[i];
        }
    }
    return v;
}
int prim(int star)
{
    vis[star] = true;
    int cnt = 0, sumlen = 0;
    for(int i = 1; i <= n; i++)
    {
        dist[i] = mp[star][i];
    }
    while(1)
    {
        int v = findmin();
        if(-1 == v)
            break;
        vis[v] = true;
        cnt++;
        sumlen += dist[v];
        for(int i = 1; i <= n; i++)
        {
            if(!vis[i] && mp[v][i] < dist[i])
            {
                dist[i] = mp[v][i];
            }
        }
    }
    if(n-1 == cnt)
        return sumlen;
    else
        return -1;
}
int main()
{
    memset(vis, 0, sizeof(vis));
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= n; j++)
        {
            if(i == j)
            {
                mp[i][j] = 0;
            }
            else
            {
                mp[i][j] = INF;
            }
        }
    }
    for(int i = 0; i < m; i++)
    {
        int t1, t2, t3;
        scanf("%d%d%d", &t1, &t2, &t3);
        mp[t1][t2] = t3;
        mp[t2][t1] = t3;
    }
    int k = prim(1);
    printf("%d\n", k);

    return 0;
}