7-10 公路村村通 (30 分)【最小生成树 模板】

版权声明:转载请注明出处,谢谢。 https://blog.csdn.net/Mercury_Lc/article/details/88408197

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;
typedef long long ll;
const int maxn = 3005;
struct node {
    int x,y,w;
    bool operator < (const node &b) const{
        return w < b.w;
    }
}s[maxn];
int fa[maxn];
int fin(int x){
    return x == fa[x] ? x :fa[x] = fin(fa[x]);
}
int Mer(int x, int y)
{
    int a = fin(x);
    int b = fin(y);
    if(a != b) {
        fa[a] = b;
        return 1;
    }
    return 0;
}
int main()
{
    ios::sync_with_stdio(0);
    int n,m;
    int u,v,w;
    cin >> n >> m;
    for(int i = 0; i < m; i ++)
    {
        cin >> u >> v >> w;
        s[i].x = u;
        s[i].y = v;
        s[i].w = w;
    }
    sort(s,s+m);
    for(int i = 0; i <= n; i ++) fa[i] = i;
    int sum = 0;
    int k = 0;
    for(int i = 0; i < m; i ++)
    {
        u = s[i].x; v = s[i].y;
        if(Mer(u,v)){
            sum += s[i].w;
            k ++;
        }
        if(k == n - 1) break;
    }
    if(k != n - 1) printf("-1\n");
    else printf("%d\n",sum);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Mercury_Lc/article/details/88408197