Acwing_1148秘密的牛奶运输【最小生成树应用: 次小生成树】

题目描述：

农夫约翰要把他的牛奶运输到各个销售点。

运输过程中，可以先把牛奶运输到一些销售点，再由这些销售点分别运输到其他销售点。

运输的总距离越小，运输的成本也就越低。

低成本的运输是农夫约翰所希望的。

不过，他并不想让他的竞争对手知道他具体的运输方案，所以他希望采用费用第二小的运输方案而不是最小的。

现在请你帮忙找到该运输方案。

注意:

• 如果两个方案至少有一条边不同，则我们认为是不同方案；
• 费用第二小的方案在数值上一定要严格小于费用最小的方案；
• 答案保证一定有解；

输入格式:

第一行是两个整数 N,M，表示销售点数和交通线路数；

接下来 M 行每行 3 个整数 x,y,z，表示销售点 x 和销售点 y 之间存在线路，长度为 z。

输出格式:

输出费用第二小的运输方案的运输总距离。

数据范围:

1≤N≤500,
1≤M≤10^4,
1≤z≤10^9,
数据中可能包含重边。

输入样例：

4 4
1 2 100
2 4 200
2 3 250
3 4 100

输出样例:

450

思路分析:

• 本题要求严格的次小生成树。我们只需要将最小生成树中某一条边替换为另一条较大的边即可，可以尝试加上每一条非树边，然后去掉多余的边，最后在所有方案中求权值最小的那个就是答案了。

代码中有求次小生成树的具体方案
AC代码:

//次小生成树
//方法一:先求最小生成树,再枚举最小生成树中的边求解,
//方法二:先求最小生成树，然后依次枚举非树边,然后将该边加入树中,同时从树中去掉一条边,使得最终的图仍是一颗树
//则一定可以求出次小生成树
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

typedef long long LL;
const int N = 505, M = 10010;
int n, m;
struct Edge {
	int a, b, w;
	bool f;;//判断是否为树边
	bool operator<(const Edge&t) {
		return w < t.w;
	}
}edge[M];

int p[N];
int dist[N][N];
int h[N], e[N * 2], w[N * 2], ne[N * 2], idx;

void add(int a, int b, int c) {
	e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}

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

void dfs(int u, int fa, int maxd, int d[]) {
	d[u] = maxd;
	for (int i = h[u]; i != -1; i = ne[i]) {
		int j = e[i];
		if (j != fa) {
			dfs(j, u, max(maxd, w[i]), d);
		}
	}
}

int main()
{
	scanf("%d%d", &n, &m);
	memset(h, -1, sizeof h);
	for (int i = 0; i < m; i++) {
		int a, b, w;
		scanf("%d%d%d", &a, &b, &w);
		edge[i] = { a,b,w };
	}

	sort(edge, edge + m);

	for (int i = 1; i <= n; i++) p[i] = i;

	LL sum = 0;
	for (int i = 0; i < m; i++) {
		int pa = find(edge[i].a), pb = find(edge[i].b), w = edge[i].w;
		if (pa != pb) {
			sum += w;//最小生成树权值总和
			p[pa] = pb;
			edge[i].f = true;
			add(edge[i].a, edge[i].b, w), add(edge[i].b, edge[i].a, w);
		}
	}
	//暴力枚举以每个点为根节点,到其他所有点的边权最大值
	for (int i = 1; i <= n; i++) dfs(i, -1, 0, dist[i]);

	LL res = 1e18;
	for (int i = 0; i < m; i++) {
		if (!edge[i].f) {//非树边
			int a = edge[i].a, b = edge[i].b, w = edge[i].w;
			if (w > dist[a][b])
				res = min(res, sum + w - dist[a][b]);
		}
	}
	printf("%lld", res);

	system("pause");
	return 0;
}