bzoj2654 二分+最小生成树

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#define N 100005
using namespace std;
int n,m,k,u
,v
,w
,c
;
struct Edge{
int u,v,w,c;
};
Edge e
;
bool cmp(const Edge &a,const Edge &b){
if(a.w<b.w)return 1;
else if(a.w==b.w){
if(a.c<b.c)return 1;
else return 0;
}
else return 0;
}
int fa
,tot=0;
int getfa(int u){
if(fa[u]==u)return u;
else return fa[u]=getfa(fa[u]);
}
bool check(int val){
tot=0;int cnt=0,ans=0;
for(int i=1;i<=n;i++)fa[i]=i;
for(int i=1;i<=m;i++){
e[i].u=u[i],e[i].v=v[i],e[i].w=w[i],e[i].c=c[i];
if(!e[i].c)e[i].w+=val;
}
sort(e+1,e+m+1,cmp);
for(int i=1;i<=m;i++){
int x=getfa(e[i].u),y=getfa(e[i].v);
if(x==y)continue;
cnt++;tot+=e[i].w;if(!e[i].c)ans++;
if(cnt==n-1)break;
fa[x]=y;
}
return ans>=k;
}
int main(){
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=m;i++){
scanf("%d%d%d%d",&u[i],&v[i],&w[i],&c[i]);
u[i]++;v[i]++;
}
int l=-110,r=110;
while(l<r){
if(r-l==1){
if(check(r))l=r;
else r=l;
}
int mid=(l+r)>>1;
if(check(mid))l=mid;
else r=mid-1;
}
printf("%d\n",tot-l*k);
}