最小生成树与建边技巧(一本通提高篇#新的开始)

直接看题面吧:

发展采矿业当然首先得有矿井,小 FF 花了上次探险获得的千分之一的财富请人在岛上挖了n口矿井,但他似乎忘记考虑的矿井供电问题……

为了保证电力的供应,小 FF 想到了两种办法:

1.在这一口矿井上建立一个发电站,费用为wi (发电站的输出功率可以供给任意多个矿井)。
2.将这口矿井与另外的已经有电力供应的矿井之间建立电网,费用为ei 。
小 FF 希望身为「NewBe_One」计划首席工程师的你帮他想出一个保证所有矿井电力供应的最小花费。

输入格式
第一行一个整数 n,表示矿井总数。

第 2行,每行一个整数wi,第 i 个数 表示在第 i 口矿井上建立发电站的费用。

接下来为一个 n*n 的矩阵 ,其中 pij 表示在第 i 口矿井和第 j 口矿井之间建立电网的费用(数据保证有 解,且pii=0,pij=pji )。

输出格式
输出仅一个整数,表示让所有矿井获得充足电能的最小花费。

样例
样例输入
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
样例输出
9

一看这道题就是一道关于最小生成树的题,但是要处理点权的问题,保证每个联通块内至少有一个发电站。
本蒟蒻由于过于笨拙没有想到正解,在考试的时候想着在最小生成树上跑DP然而失败了。。。

接下来就讲一下正解吧,新建一个超级发电站,这个点和每一个点连一条权值为wi的点,再在这个图上重新跑一边最小生成树就可以了。这个证明就跳过了吧,正确性自己仔细想一想就知道了。

那么直接上代码吧:

#include <bits/stdc++.h>
#define maxn 310
using namespace std;

int n, fa[maxn], val[maxn], cnt = 0, ans = 0, tot = 0;
struct edge {
    int to, from, val;
} e[maxn * maxn];

inline void add(int x, int y, int z) {
    e[++cnt].to = y;
    e[cnt].from = x;
    e[cnt].val = z;
}
inline bool cmp(edge a, edge b) { return a.val < b.val; }

inline int find(int x) {
    if (x == fa[x])
        return x;
    return fa[x] = find(fa[x]);
}
int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n + 1; ++i) fa[i] = i;
    for (int i = 1; i <= n; ++i) scanf("%d", &val[i]);
    for (int i = 1; i <= n; ++i) {
        for (int j = 1; j <= n; ++j) {
            int x;
            scanf("%d", &x);
            if (i == j)
                continue;
            add(i, j, x);
        }
    }
    for (int i = 1; i <= n; ++i) add(i, n + 1, val[i]), add(n + 1, i, val[i]);
    sort(e + 1, e + 1 + cnt, cmp);
    for (int i = 1; i <= cnt; ++i) {
        int x = find(e[i].from);
        int y = find(e[i].to);
        if (x == y)
            continue;
        fa[x] = y;
        tot++;
        ans += e[i].val;
        if (tot == n)
            break;
    }
    printf("%d", ans);
    return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值