新学的科技。设\(f(x)\)为选\(x\)条白色边的时候的最小生成树权值和,那么可以猜到它应该是一个下凸函数的形式。
如图,图中\(x\)坐标表示选的白色边条数,\(y\)坐标表示获得的权值,那么我们就可以把\(f(x)\)在这个图上大致表示出来。我们现在并不清除\(x\)和\(y\),所以可以二分一下和这个凸函数相切直线的斜率。设这个直线为\(y = kx + b\),那么对于一个固定的\(x\),截距最小的时候,就是与函数相切的时候嘛,也是答案最优的时候。
我们把这个直线转化成\(y - kx = b\)的形式。由于不清楚会选用几条边,所以可以提前给每一条白色边都减去一个\(k\),这样不管选几条边其影响都可以被直接统计。也就是说我们现在就可以忽略选几条边的问题直接去最小化截距\(b\)了。在最小化截距的同时我们对\(y\)的值和\(x\)的值做一个记录,这样就可以做出应该取用左区间还是右区间的判定啦。
#includeusing namespace std;const int N = 50000 + 5;const int M = 100000 + 5;#define pii pair #define mp(x,y) make_pair (x, y)struct Len { int u, v, w, c; void read () { cin >> u >> v >> w >> c; } bool operator < (Len rhs) const { return w == rhs.w ? c < rhs.c : w < rhs.w; }}L[M];int n, m, k, fa[N];int find (int x) { return x == fa[x] ? x : fa[x] = find (fa[x]);}pii Kruskal () { for (int i = 0; i < n; ++i) fa[i] = i; sort (L, L + m); int cnt = 0, ret = 0, wht = 0; for (int i = 0; i < m; ++i) { int fu = find (L[i].u); int fv = find (L[i].v); if (fu != fv) { cnt += 1; fa[fu] = fv; ret += L[i].w; wht += L[i].c == 0; } if (cnt == m - 1) break; } return mp (wht, ret);} signed main () {// freopen ("data.in", "r", stdin); cin >> n >> m >> k; for (int i = 0; i < m; ++i) { L[i].read (); } int l = -150, r = 150, ans = 0; while (l < r) { int mid = (l + r) >> 1; for (int i = 0; i < m; ++i) { if (L[i].c == 0) { // 白色 L[i].w -= mid; } } pii ret = Kruskal ();// cout << "l = " << l << " r = " << r << " mid = " << mid << " ret = (" << ret.first << ", " << ret.second << ")" << endl; if (ret.first >= k) { r = mid; ans = ret.second + mid * k; } else { l = mid + 1; } for (int i = 0; i < m; ++i) { if (L[i].c == 0) { L[i].w += mid; } } }// cout << l << " " << r << endl; cout << ans << endl;}