【bzoj 2654】tree（参数二分+最小生成树）

传送门biu~
因为要固定白色边的数量，所以我们可以让“选取一条白色边”有一个代价。显然，随着“选取一条白色边”代价的降低，最小生成树上白色边的数目一定是不降的。因此我们可以二分这个代价，每次验证时让所有白色边权加上这个代价，然后跑最小生成树，最后再将这些代价减去。
因为保证数据合法，所以假设代价为mid时用的白色边数大于need，代价为mid+1时白色变数小于need，一定可以在代价为need时用一些黑色边代替白色边。具体操作时只需要在减去多余代价时减去代价*need即可。

#include<bits/stdc++.h>
#define N 50005
using namespace std;
struct Edge{int s,t,v,c;}e[N<<1];
inline bool cmp(Edge a,Edge b){return a.v<b.v || a.v==b.v && a.c<b.c;}
int n,m,need,fa[N];
inline int search(int x){return fa[x]==x ? x: fa[x]=search(fa[x]);}
inline int binary_search(){
    int l=-101,r=101,res;
    while(l<=r){
        int mid=l+r>>1;
        for(int i=0;i<m;++i) if(!e[i].c) e[i].v+=mid;
        sort(e,e+m,cmp);
        for(int i=0;i<n;++i) fa[i]=i;
        int ans=0,wh=0;
        for(int x,y,num=0,i=0;i<m;++i){
            x=search(e[i].s),y=search(e[i].t);
            if(x^y){
                fa[x]=y;
                ans+=e[i].v;
                if(!e[i].c) ++wh;
                if(++num==n-1) break;
            }
        }
        if(wh>=need){
            l=mid+1;
            res=ans-mid*need;
        }
        else r=mid-1;
        for(int i=0;i<m;++i) if(!e[i].c) e[i].v-=mid;
    }
    return res;
}
int main(){
    scanf("%d%d%d",&n,&m,&need);
    for(int i=0;i<m;++i)    scanf("%d%d%d%d",&e[i].s,&e[i].t,&e[i].v,&e[i].c);
    printf("%d\n",binary_search());
    return 0;
}