POJ 2421 Constructing Roads (MST)

题中给了n*n的矩阵,值是i点到j点的建边的花费,其中已经建好了m条边,题中求还需要花费多少才能使该图连通

思路:Kruskal更好做(并查集)。初始化之后,将m条边依次加入并查集,只要能合并及时合并。

/***********************************************
 * Author: fisty
 * Created Time: 2015/2/28 14:04:54
 * File Name   : D.cpp
 *********************************************** */
#include <iostream>
#include <cstring>
#include <deque>
#include <cmath>
#include <queue>
#include <stack>
#include <list>
#include <map>
#include <set>
#include <string>
#include <vector>
#include <cstdio>
#include <bitset>
#include <algorithm>
using namespace std;
#define Debug(x) cout << #x << " " << x <<endl
#define Memset(x, a) memset(x, a, sizeof(x))
const int INF = 0x3f3f3f3f;
typedef long long LL;
typedef pair<int, int> P;
#define FOR(i, a, b) for(int i = a;i < b; i++)
#define MAX_N 10010
int n;
int cnt;
struct edge{
    int u;
    int v;
    int cost;
}G[MAX_N];

void addedge(int u, int v, int cost){
    G[cnt].u = u;
    G[cnt].v = v;
    G[cnt].cost = cost;
    cnt++;
}
bool cmp(edge e1, edge e2){
    return e1.cost < e2.cost;
}
int par[MAX_N];
void init(){
    for(int i = 0;i <= n; i++){
        par[i] = i;
    }
}
int find(int x){
    if(x == par[x]) return x;
    return par[x] = find(par[x]);
}
void unio(int x, int y){
    x = find(x);
    y = find(y);
    if(x != y){
        par[x] = y;
    }
}
int kurskal(){
    sort(G, G + cnt, cmp);
    int ans = 0;
    for(int i = 0;i < cnt; i++){
        edge e = G[i];
        if(find(e.u) != find(e.v)){
            unio(e.u, e.v);
            ans += e.cost;
        }
    }
    return ans;
}


int main() {
    //freopen("in.cpp", "r", stdin);
    cin.tie(0);
    ios::sync_with_stdio(false);
    cin >> n;
    cnt = 0;
    FOR(i, 1, n+1){
        FOR(j, 1, n+1){
            int cost;
            cin >> cost;
            if(i != j)
                addedge(i, j, cost);
        }
    }
    init();
    int Q;
    cin >> Q;
    FOR(i, 0, Q){
        int u, v;
        cin >> u >> v;
        unio(u, v);
    }
    int ans = kurskal();
    cout << ans << endl;
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值