bzoj 2654 tree （二分 + 最小生成树）

Description

给你一个无向带权连通图，每条边是黑色或白色。让你求一棵最小权的恰好有need条白色边的生成树。
题目保证有解。

Input

第一行V,E,need分别表示点数，边数和需要的白色边数。
接下来E行,每行s,t,c,col表示这边的端点(点从0开始标号)，边权，颜色(0白色1黑色)。

Output

一行表示所求生成树的边权和。
V<=50000,E<=100000,所有数据边权为[1,100]中的正整数。

Sample Input

2 2 1

0 1 1 1

0 1 2 0

Sample Output

2

题解

woc打完代码之后发现和大佬的代码相似度99%……可怕
考试的时候完全没有思路，考完之后评讲：二分？那是什么?
果然我还是见识得太少了。第一次见二分可以这样用来着。
二分一个权值，使白边的边权集体增加一个mid；这样做能的目的是调整每一次最小生成树中白边的数量。显然白边的边权加得越多，对于白边的选择就越少，这是单调递增的。而即使白边加上了权值，只要得出的树的形态是固定的，那么它的权值就还能再减回去。答案就是加上权值跑出来白边的边数为need时的答案-need*mid。这样就能保证正确性了。
至于还有一种情况：当存在一个mid，使跑出来的树白边的比need多，而mid+1却比need少，这时考虑树中一定存在黑边，且这些黑边的权值和加上mid之后的白边的权值相同（因为保证有解）。那么就假想可以用这些黑边替换掉多余的白边就可以了。

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int N = 1000020;struct Node {    int u, v, w, c, next;} e[N];int n, m, need, now, ans, num = 0, head[N], fa[N];void add(int u, int v, int w, int c) {    num ++;    e[num].u = u; e[num].v = v; e[num].c = c;    e[num].w = w; e[num].next = head[u]; head[u] = num;}bool cmp(const Node &a, const Node &b) {    if(a.w == b.w) return a.c < b.c;    return a.w < b.w;}int find(int x) {    if(x == fa[x]) return x;    return fa[x] = find(fa[x]);}bool check() {    int tot = 0, sum = 0; now = 0;    sort(e + 1, e + m + 1, cmp);    for(int i = 1; i <= n; i ++) fa[i] = i;    for(int i = 1; i <= m; i ++) {        int fx = e[i].u, fy = e[i].v;        if(fx == fy) continue;        fa[fy] = fx;         now += e[i].w; sum += ! e[i].c;        if(++ tot == n - 1) break;    }    if(sum >= need) return true;    else return false;}int main() {    scanf("%d %d %d", &n, &m, &need);    for(int i = 1; i <= m; i ++) {        int u, v, w, c;        scanf("%d %d %d %d", &u, &v, &w, &c);        add(u, v, w, c);    }    int l = - 101, r = 101, mid;    while(l <= r) {        mid = l + r >> 1;        for(int i = 1; i <= m; i ++)            if(! e[i].c) e[i].w += mid;        if(check()) l = mid + 1, ans = now - need * mid;        else r = mid - 1;        for(int i = 1; i <= m; i ++)            if(! e[i].c) e[i].w -= mid;    }    printf("%d\n", ans);    return 0;}