题目戳这里:http://www.lydsy.com/JudgeOnline/problem.php?id=2654
我们知道对于一个最小生成树而言,它的白边权值越大,那么生成树中的白边个数越少,反之亦然。
所以我们每次二分一个值,给白边加上它,在做最小生成树判断是否大于所需要的,然后就可以二分来搞了。
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#define N 100005
using namespace std;
int n,m,k,u[N],v[N],w[N],c[N];
struct Edge{
int u,v,w,c;
};
Edge e[N];
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[N],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);
}