传送门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;
}