BZOJ2654: tree

二分 最小生成树

题目传送门

这种二分貌似和WQS二分有点像。

二分一个给白色边的边权$x$，当最后求得的生成树白色边$\geq need$时更新$ans=sum-need*x$并上调边权使得下次选择白色边的数目减少，否则下调边权。正确性YY一下应该比较显然。

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 50005
#define F inline
using namespace std;
struct edge{ int x,y,z,f,id; }ed[N<<1];
int n,m,k,q,sum,fa[N],h[N],w[N<<1];
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
F bool cmp(edge a,edge b){ return a.z==b.z?a.f<b.f:a.z<b.z; }
int findfa(int x){ return x==fa[x]?x:fa[x]=findfa(fa[x]); }
F bool pd(int x){
    for (int i=1;i<=n;i++) fa[i]=i;
    for (int i=1;i<=m;i++)
        ed[i].z=w[ed[i].id]+x*(ed[i].f^1);
    sort(ed+1,ed+m+1,cmp),sum=0; int num=0;
    for (int i=1,fx,fy;i<=m;i++){
        if ((fx=findfa(ed[i].x))!=(fy=findfa(ed[i].y)))
            fa[fx]=fy,sum+=ed[i].z,num+=(ed[i].f^1);
    }
    return num>=q;
}
int main(){
    n=_read(),m=_read(),q=_read();
    for (int i=1,x,y,z,f;i<=m;i++){
        x=_read()+1,y=_read()+1,z=_read(),f=_read();
        w[i]=z,ed[++k]=(edge){x,y,z,f,i};
    }
    int l=-105,r=105,d,ans=0;
    while (l<=r)
        pd(d=l+r>>1)?ans=sum-q*d,l=d+1:r=d-1;
    return printf("%d\n",ans),0;
}