[bzoj2654] tree

　　一开始以为先取need条最短的白边就行了。。然而那样子的话可能图根本没法联通= =

　　网上题解讲的挺清晰的。。就是二分把全部白边加上mid，然后看mst里面有多少条白边。有need条白边的时候再把加上的值减去，就是答案了。

　　但可能出现取不了need条白边的情况（二分mid取到>need条，二分mid+1时取到<need条）

　　这时候就是因为图中有一些黑边和白边权值相同。。那显然我们取的是黑边还是白边是可以替换的。。

　　所以取到>=need条白边的时候就可以更新答案。注意权值相同的话优先先取白边

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100233;
struct zs{
    int u,v,dis;
}e[maxn],e1[maxn];int t,t1;
int fa[maxn],id[maxn];
int i,j,k,n,m,ans,a,b,c,d,now,l,r,mid;

int ra;char rx;
inline int read(){
    rx=getchar(),ra=0;
    while(rx<'0'||rx>'9')rx=getchar();
    while(rx>='0'&&rx<='9')ra*=10,ra+=rx-48,rx=getchar();return ra;
}
bool cmp(int a,int b){return e[a].dis==e[b].dis?a<b:e[a].dis<e[b].dis;}
inline int getfa(int x){return fa[x]!=x?fa[x]=getfa(fa[x]):x;}
inline int get(int x){
    register int i,sm=1,num0=0;now=0;
    for(i=1;i<=t;i++)e[i].dis+=x;
    for(i=1;i<=n;i++)fa[i]=i;
    sort(id+1,id+1+m,cmp);
//    for(i=1;i<=m;i++)printf("! %d   %d--%d %d    %d %d
",id[i],e[id[i]].u,e[id[i]].v,e[id[i]].dis,getfa(e[id[i]].u),getfa(e[id[i]].v));
    for(i=1;i<=m&&sm<n;i++)
        if((a=getfa(e[id[i]].u))!=(b=getfa(e[id[i]].v)))
            fa[a]=b,sm++,
            num0+=id[i]<=t,
            now+=e[id[i]].dis;//,printf("  %d ",id[i]);
    
    for(i=1;i<=t;i++)e[i].dis-=x;
//    printf("   x:%d num0:%d now:%d   sm:%d
",x,num0,now,sm);
    return num0;
}
int main(){register int i;
    n=read(),m=read(),k=read();
    for(i=1;i<=m;i++){
        a=read()+1,b=read()+1,c=read(),d=read();
        if(d)e1[++t1]=(zs){a,b,c};
        else e[++t]=(zs){a,b,c};
    }
    for(i=t+1;i<=m;i++)e[i]=e1[i-t];
    for(i=1;i<=m;i++)id[i]=i;
    l=-101,r=101;ans=1002333333;
    while(l<=r){
        mid=(l+r)/2;//printf("    %d -- %d
",l,r);
        if(get(mid)>=k)ans=now-mid*k,l=mid+1;
            else r=mid-1;
    }
    printf("%d
",ans);
    return 0;
}