原题链接
解题思路:不会^ v ^
C++代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
const int M=100005;
int n,m,k,tot,cnt;
int f[M];
struct edge{int u,v,w,c;}e[M*2];
int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int fd(int x){return f[x]==x?x:f[x]=fd(f[x]);}
bool cmp(edge a,edge b){return a.w<b.w||(a.w==b.w&&a.c<b.c);}
bool check(int x)
{
tot=cnt=0;
for(int i=1;i<=n;i++) f[i]=i;
for(int i=1;i<=m;i++) if(!e[i].c) e[i].w+=x;
sort(e+1,e+m+1,cmp);
for(int i=1;i<=m;i++)
{
int p=fd(e[i].u),q=fd(e[i].v);
if(p!=q){
f[p]=q;tot+=e[i].w;
if(!e[i].c) cnt++;
}
}
for(int i=1;i<=m;i++) if(!e[i].c) e[i].w-=x;
return cnt>=k;
}
void init()
{
n=read();m=read();k=read();
for(int i=1;i<=m;i++)
e[i].u=read(),e[i].v=read(),e[i].w=read(),e[i].c=read(),e[i].u++,e[i].v++;
}
void work()
{
int sumv,l=-105,r=105;
while(l<=r)
{
int mid=(l+r)>>1;
if(check(mid)) l=mid+1,sumv=tot-k*mid;
else r=mid-1;
}
printf("%d",sumv);
}
int main()
{
init();
work();
return 0;
}
参考博客