7-8 畅通工程之最低成本建设问题

7-8 畅通工程之最低成本建设问题

某地区经过对城镇交通状况的调查，得到现有城镇间快速道路的统计数据，并提出“畅通工程”的目标：使整个地区任何两个城镇间都可以实现快速交通（但不一定有直接的快速道路相连，只要互相间接通过快速路可达即可）。现得到城镇道路统计表，表中列出了有可能建设成快速路的若干条道路的成本，求畅通工程需要的最低成本。

输入格式:

输入的第一行给出城镇数目N (1<N≤1000)和候选道路数目M≤3N；随后的M行，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号（从1编号到N）以及该道路改建的预算成本。

输出格式:

输出畅通工程需要的最低成本。如果输入数据不足以保证畅通，则输出“Impossible”。

输入样例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

输出样例1:

12

输入样例2:

5 4
1 2 1
2 3 2
3 1 3
4 5 4

输出样例2:

Impossible

解

这道题跟那个公路村村通是一类题目， 最小生成树问题

上code

#include<bits/stdc++.h>

using namespace std;
#define M 3003
struct s{
    int x, y;
    int weight;
}t[M];

int parent[M];

bool cmp(s &a, s &b)
{
    return a.weight < b.weight;
}

int findparent(int a)
{
    while(a != parent[a])
        a = parent[a];
    return a;
}

int Kruskal(int n, int m)
{
    int num = 0, sum = 0;
    for(int i = 0; i < m; i++)
    {
        int a = findparent(t[i].x);
        int b = findparent(t[i].y);
        if(a != b)  // 若 a != b 则说明目前生成树还没有生成环路
        {
            parent[a] = b;  /// 表示此顶点已在生成树集合中
            num++;
            sum += t[i].weight;
        }
        if(num == n - 1)
            break;
    }
    if(num == n - 1)
        return sum;
    return -1;
}

int main()
{
    int n, m;
    cin >> n >> m;
    for(int i = 0; i < n; i++)
        parent[i] = i;
    for(int i = 0; i < m; i++)
    {
        cin >> t[i].x >> t[i].y >> t[i].weight;
    }
    sort(t, t + m, cmp);
    int res = Kruskal(n, m);
    if(res == -1)
        cout << "Impossible" << endl;
    else
        cout << res << endl;
    return 0;
}