AcWing 1148. 秘密的牛奶运输 (求严格次小生成树)

Description

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

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

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

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

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

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

注意：

• 如果两个方案至少有一条边不同，则我们认为是不同方案

• 费用第二小的方案在数值上一定要严格大于费用最小的方案

• 答案保证一定有解

Input

第一行是两个整数 n, m ( 1 ≤ n ≤ 500 ,  1 ≤ m ≤ 104 )，表示销售点数和交通线路数；

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

Output

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

Sample Input

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

Sample Output

450

思路：先求出最小生成树，标记树边和非树边。枚举添加每一条非树边并对应删去一条树边后得到的新生成树，从中更新出严格次小生成树的值。

细节剖析：记添加一条权值为w的非树边，其连接的两个端点为a,b，则对应删除的边应该为最小生成树中a到b路径上小于w的一条最大的边，才能使得新生成树的权值尽可能小(如果路径中所有的边权值都等于w，即找不到对应删除的边，则替换该非树边后的新生成树权重不变，不符合严格次小生成树，而题目保证有解，因此跳过该条非树边，继续枚举其它非树边即可) 

代码如下：

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

#define N 520
#define M 10020

struct node {
	int a, b, w;
	bool inmst;//判断是否为树边
	bool operator < (const node& W)const {
		return w < W.w;
	}
}edge[M];

int h[N], e[2 * M], w[2 * M], ne[2 * M], k;
void add(int a, int b, int c) {
	e[k] = b, w[k] = c, ne[k] = h[a], h[a] = k++;
	e[k] = a, w[k] = c, ne[k] = h[b], h[b] = k++;
}

int n, m, idx;
int p[N];
int find(int x) {
	if (x == p[x])return x;
	return p[x] = find(p[x]);
}

int max1[N][N], max2[N][N];
void dfs(int a, int b, int fa, int w1, int w2) {
	max1[a][b] = w1; max2[a][b] = w2;
	for (int i = h[b]; i != -1; i = ne[i]) {
		if (e[i] == fa)continue;
		int t1 = w1, t2 = w2, v = w[i];
		if (v > t1) { t2 = t1, t1 = v; }
		else if (v > t2 && v < t1)t2 = v;
		dfs(a, e[i], b, t1, t2);
	}
}


long long ans, sum;

int main() {

	cin >> n >> m;
	for (int i = 0; i <= n; i++)p[i] = i;
	int a, b, w;
	for (int i = 0; i < m; i++) {
		cin >> a >> b >> w;
		edge[idx++] = { a,b,w,false };
	}
	sort(edge, edge + idx);

	memset(h, -1, sizeof h);
	for (int i = 0; i < idx; i++) {
		int a = edge[i].a, b = edge[i].b, w = edge[i].w;
		int fa = find(a), fb = find(b);
		if (fa == fb)continue;
		add(a, b, w);//将最小生成树建图
		p[fa] = fb;
		sum += w;//最小生成树权重
		edge[i].inmst = true;
	}

	//计算在最小生成树上各点间路径的最大值和次大值
	for (int i = 1; i <= n; i++)dfs(i, i, -1, 0, 0);

	ans = 1e20;
	for (int i = 0; i < idx; i++) {//枚举替换每一条非树边后新生成树的最小值
		if (edge[i].inmst)continue;
		int a = edge[i].a, b = edge[i].b, v = edge[i].w;
		//如果路径最大值小于v,则替换该树边，否则替换次大值树边
		if (v > max1[a][b])ans = min(ans, sum - max1[a][b] + v);
		//如果路径最大值边等于v，则替换次大值边
		//如果路径中所有边权等于v，没有次大值，则次大值没有被更新过为0，因此需要特判次大值是否为0
		else if (v > max2[a][b] && max2[a][b])ans = min(ans, sum - max2[a][b] + v);
	}

	cout << ans << endl;

	return 0;

}