公路村村通 并查集简单应用

现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。
输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。

思路: 将两节点与花费的结构体按花费升序排序,遍历每个结构体,若不互相连通,将两点加入并查集,并更新花费。最后遍历任意两点判断是否存在不通畅的情况

代码:

#include<iostream>
#include<queue>
#include<stack>
#include<cstring>
#include<stdlib.h>
#include<cstdio>
#include<iostream>
#include<algorithm>

#define LL long long
using namespace std;
#define MAXN 3001

struct node{
    int f, s, cost;
};

bool cmp(node a, node b){
    return a.cost < b.cost;
}

node c[MAXN];
int F[MAXN];

void init(int n){
    for(int i = 1;  i <= n; i++)
        F[i] = i;
}

int find(int x){
    if(F[x] != x)
        F[x] = find(F[x]);
    return F[x];
}

bool same(int x, int y){
    if(find(x) == find(y))
        return true;
    return false;
}

void unionn(int x, int y){
    int t1 = find(x);
    int t2 = find(y);
    if(t1 != t2){
        F[t1] = t2;
    }
}

//int w[MAXN][MAXN];
int main(){
    ios :: sync_with_stdio(false);
    int N, M;
    cin >> N >> M;
    init(N);
    int sum = 0;
    for(int i = 0; i < M; i++){
        cin >> c[i].f >> c[i].s >> c[i].cost;
        //w[c[i].f][c[i].s] = c[i].cost;
        //unionn(c[i].f, c[i].s);
    }
    sort(c, c+M, cmp);


    for(int i = 0; i < M; i++){
        if(!same(c[i].f, c[i].s)){
            sum += c[i].cost;
            unionn(c[i].f, c[i].s);
        }
    }
    for(int i = 1; i <= N; i++){
        for(int j = i+1; j <= N; j++){
            if(!same(i, j)){
                cout << -1 << endl;
                return 0;
            }
        }
    }
    cout << sum << endl;
    //system("pause");
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值