蓝桥-算法训练 ALGO-6 安慰奶牛【最小生成树】

题目链接

题目描述：

原题的描述真的烂。
题意是这样的：给一张图，每个点和边都有权值，经过点和边的花费为权值的大小，现在要将这个图去掉一些边使其成为一颗生成树(题目只给了图)，求从生成树的一个点 s (题目没给)出发，访问其他所有的点至少一次并返回 s 的最少花费。

---

题解：

这一看就是与最小生成树有关的题，因为题目要求从 s 出发最终还要回到 s，所以我们易知如果经过一条边，那么就要从这条边再回来(边要经过两次)，因为是求最小生成树，我们得想办法将结点的权值合并到边上，对于一条新的从 u -> v 的路径，新增加的路径我们只算结点 v 带来的贡献 $c_v$ ，那么对于一个非起始点 x，一开始它的角色是必定是 v，所以先加上 $c_x$，然后要从这个点向其他点走，所以它的角色就变成 u 了，因为从 u 出发到 v 必定还要从 v 返回到 u (这时候 x 就变成 v 了，又要加上 $c_x$)，所以走过一条边，就必定还要花费这条边两个端点的权值，设边的权值为 L，点的权值为 c，边的两个端点分别为 u 和 v，那么一条边的权值就是 $W=2\ast L+c_u+c_v$，所以根据这个权值求最小生成树的取值 sum 即可，因为在上述的(u -> v)角色中，起点 s 一开始就是 u，所以它的权值没有被加上(因为我们只加角色为 v 的结点的权值)，所以最终还要再加上它，又因为最小生成树的权值是固定的，所以最终答案就是$ans=sum+min(c_i)$。

---

AC Codes：

#include <iostream>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
//#include <unordered_set>
//#include <unordered_map>
#include <deque>
#include <list>
#include <iomanip>
#include <algorithm>
#include <fstream>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <cstdlib>
//#pragma GCC optimize(2)
using namespace std;
typedef long long ll;
//cout << fixed << setprecision(2);
//cout << setw(2);
const int N = 1e4 + 6, M = 1e9 + 7, INF = 0x3f3f3f3f;

int n, m, c[N], fa[N], d[N];
struct Edge {
    int u, v, w;
    bool operator < (const Edge &other) const {
        return w < other.w;
    }
}e[100005];

int find(int x) {
    return x == fa[x] ? x : find(fa[x]);
}

int kruskal() {
    ll res = 0;
    for (int i = 1; i <= n; i++) fa[i] = i;
    for (int i = 0; i < m; i++) {
        int x = find(e[i].u), y = find(e[i].v);
        if (x != y) {
            fa[y] = x;
            res += e[i].w;
        }
    }
    return res;
}

int main() {
    //freopen("/Users/xumingfei/Desktop/ACM/test.txt", "r", stdin);
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    cin >> n >> m;
    for (int i = 1; i <= n; i++) cin >> c[i];
    int u, v, w;
    for (int i = 0; i < m; i++) {
        cin >> e[i].u >> e[i].v >> e[i].w;
        e[i].w += e[i].w + c[e[i].u] + c[e[i].v];
    }
    sort(e, e + m);
    int mn = 1e9;
    for (int i = 1; i <= n; i++) mn = min(mn, c[i]);
    cout << mn + kruskal() << '\n';
    return 0;
}