Luogu P4716 【模板】最小树形图
不知道有什么用的一个东西。本来不打算再大量扩知识点了但还是学一下好了,反正也不难。
原理:树上父亲唯一,每次选最短的父边。
此时会有两类情况:
-
就这样正常连下去,这样我们就得到了一个尽可能小的树形图。
-
成环。这种情况下我们需要拆掉环里的一条边换成其他的边。
我们记录一下到达每个点的最短父边权值是多少。对于成环的情况,可以先把环里面所有边的权值选上,把环里面的所有点看成一个。然后等到有其它外来的边连进来的时候,再选一个最小的外来边,去掉环里面原先所暂时使用的边,换成外来的那个,就这样一直求解直到不再有环。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 100 + 5;
const int M = 10000 + 5;
const ll INF = 0x3f3f3f3f;
int n, m, r;
struct edge {int u, v, w;} e[M];
int fa[N], id[N], top[N], minw[N];
ll get_ans (int n, int m) {
ll ans = 0;
while (true) {
int cnt = 0;
for (int i = 1; i <= n; ++i) {
id[i] = top[i] = 0; minw[i] = INF;
}
for (int i = 0; i < m; ++i) {
if (e[i].u != e[i].v && e[i].w < minw[e[i].v]) {
fa[e[i].v] = e[i].u;
minw[e[i].v] = e[i].w;
}
}
minw[r] = 0;
for (int i = 1; i <= n; ++i) {
if (minw[i] == INF) return -1;
ans += minw[i];
int u = i;
while (u != r && top[u] != i && !id[u]) {
top[u] = i;
u = fa[u];
}
if (u != r && !id[u]) {
id[u] = ++cnt;
for (int v = fa[u]; v != u; v = fa[v]) id[v] = cnt;
}
}
if (cnt == 0) return ans;
for (int i = 1; i <= n; ++i) {
if (!id[i]) id[i] = ++cnt;
}
for (int i = 0; i < m; ++i) {
int prew = minw[e[i].v];
e[i].u = id[e[i].u];
e[i].v = id[e[i].v];
if (e[i].u != e[i].v) {
e[i].w -= prew;
}
}
n = cnt; r = id[r];
}
}
int main () {
cin >> n >> m >> r;
for (int i = 0; i < m; ++i) {
static int u, v, w;
cin >> u >> v >> w;
e[i] = (edge) {u, v, w};
}
cout << get_ans (n, m) << endl;
}
以及非常感谢 @旋转卡壳 的代码。仅仅是读注释就可以快速理解整个算法的流程。(虽然代码不加空格\(www\))