【CF1468J】J. Road Reform（最小生成树，贪心）

重建路

题意：

• 给一个无向图，要求用最少的边连通所有的点，同时图中最大的边权正好是 K 。
• 你可以花费一次操作，将一条边的边权 - 1 或 + 1。问你满足要求的最少花费是多少。

思路：

• 把所有边分成 小于 K 和 大于等于 K，分情况讨论：

1. 只用 小于 K 的边即可连通所有点

• 先连通，建出这颗树。
• 根据树的性质，我们可以尝试任意挑选一条边，替换树中的边（选树中的边替换也是可以的，相当于不变，花费操作这条边）。要求最少花费，自然这条边要离 K 尽可能近。

2. 只用 小于 K 的边不可连通所有点

• 因为还需要边来建树，这些边的权值都是 大于等于 K 的。贪心的想，我们要选尽可能小的大于等于 K 的边。因为最大边权不能超过 K ，所以这些边都要通过操作减权使得边权变成 K。

$Code:$

#include<bits/stdc++.h>
#include<unordered_map>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define sca scanf
#define pri printf
#define forr(a,b,c) for(int a=b;a<=c;a++)
#define rfor(a,b,c) for(int a=b;a>=c;a--)
#define all(a) a.begin(),a.end()
#define oper(a) (operator<(const a& ee)const)
#define endl "\n"
using namespace std;

typedef long long ll;
typedef pair<int, int> PII;

double DNF = 1e17;
const int N = 200010, M = 400010, MM = 110;
int INF = 0x3f3f3f3f, mod = 1e9 + 7;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int n, m, k, T, S, D, K;
int p[N];
struct edge
{
	int a, b, w;
	bool oper(edge) { return w < ee.w; }
}ed[M];

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

void solve() {
	cin >> n >> m >> k;
	forr(i, 1, n)p[i] = i;

	forr(i, 1, m) {
		int a, b, w;
		cin >> a >> b >> w;
		ed[i] = { a,b,w };
	}
	sort(ed + 1, ed + m + 1);//贪心建最小生成树

	int cnt = 0;
	forr(i, 1, m) {
		if (ed[i].w >= k)continue; //先用小于 K 的边
		int a = ed[i].a, b = ed[i].b;
		a = find(a), b = find(b);
		if (a ^ b) {
			p[a] = b;
			cnt++;
		}
	}

	if (cnt == n - 1) { //小于 K 的边已经能成树的情况
		int mi = INF;
		forr(i, 1, m)mi = min(mi, abs(ed[i].w - k));
		cout << mi << endl;
		//选一条离 K 最近的边增到 K 即可
	}
	else { //不能的情况
		// 我们继续建最小生成树
		ll sum = 0;
		forr(i, 1, m) {
			if (ed[i].w < k)continue;
			int a = ed[i].a, b = ed[i].b;
			a = find(a), b = find(b);
			if (a ^ b) {
				p[a] = b;
				sum += ed[i].w - k;//大于等于的边都要减成 K
			}
		}
		cout << sum << endl;
	}
}

int main() {
	cinios;
	cin >> T;
	while (T--)solve();
	return 0;
}
/*
*/