08-图7 公路村村通

Posted 2020-09-18 whileskies的博客

题目：

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

输入格式:

输入数据包括城镇数目正整数N（≤1000）和候选道路数目M（≤3N）；随后的M行对应MM条道路，每行给出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 <iostream>
#include <algorithm>
using namespace std;
int w[5500];
int u[5500], v[5500];
int r[5500];
int p[1100];

int cmp(const int a, const int b)
{
    return w[a] < w[b];
}

int Find(int x)
{
    return p[x] == x ? x : p[x] = Find(p[x]);
}

int Kuskal(int n, int m)
{
    int ans = 0;
    int cnt = 0;
    for (int i = 1; i <= n; i++)
        p[i] = i;
    for (int i = 1; i <= m; i++)
        r[i] = i;
    sort(r+1, r+m+1, cmp);
    for (int i = 1; i <= m; i++)
    {
        int e = r[i];
        int x = Find(u[e]);
        int y = Find(v[e]);
        if (x != y) {
            ans += w[e];
            cnt++;
            p[x] = y;
        }
    }
   if (cnt < n-1)
        return -1;
   else
        return ans;
}

int main()
{
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= m; i++)
        cin >> u[i] >> v[i] >> w[i];
    int ans = Kuskal(n, m);
    if (ans == 0 || ans == -1)
        cout << "-1";
    else
        cout << ans;


    return 0;
}