bzoj2654 tree 二分+最小生成树

Description

---

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

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

Solution

---

这是我在ubuntu上写出的一道题目，一半是vim写的另一半是sublime写的
可以想到把全体白边加上或减去一个权值来控制最小生成树中白边的数量，二分这个权值即可

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))

const int N=200005;
const int E=800005;

struct edge {int x,y,w,c;} e[E];

int fa[N],n,m,mid,ans,need;

int get_father(int x) {
    if (fa[x]==x) return x;
    return fa[x]=get_father(fa[x]);
}

bool merge(int x,int y) {
    int fx=get_father(x);
    int fy=get_father(y);
    if (fx==fy) return false;
    fa[fx]=fy;
    return true;
}

bool cmp(edge a,edge b) {
    int w1=a.w,w2=b.w;
    if (a.c==0) w1-=mid;
    if (b.c==0) w2-=mid;
    return w1<w2;
}

int mst() {
    int ret=0,cnt=0; ans=0;
    rep(i,1,n) fa[i]=i;
    std:: sort(e+1,e+m+1,cmp);
    rep(i,1,m) {
        if (cnt==m-1) break;
        if (merge(e[i].x,e[i].y)) {
            ans+=e[i].w;
            ret+=!e[i].c;
            cnt++;
        }
    }
    return ret;
}

int main(void) {
    scanf("%d%d%d",&n,&m,&need);
    rep(i,1,m) {
        scanf("%d%d%d%d",&e[i].x,&e[i].y,&e[i].w,&e[i].c);
        e[i].x++; e[i].y++;
    }
    int l=-1000,r=1000;
    while (l<=r) {
        mid=(l+r)>>1;
        int ret=mst();
        if (ret==need) break;
        if (ret>need) r=mid-1;
        else l=mid+1;
    }
    printf("%d\n",ans);
    return 0;
}