题目链接:Bob’s Problem
题意:就是给你一些白边和一些黑边,让你随便选,但是白边不超过k条。
很明显黑边能选就选,然后白边贪心从大往小,先从大到小让整个图连通,然后从大到小选没选过的即可。
AC代码:
#pragma GCC optimize(2)
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e4+10,M=5e5+10;
int T,n,m,k,res,f[N],vis[M],flag;
struct node{int u,v,w;}; vector<node> g;
int cmp(node a,node b){return a.w>b.w;}
int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
signed main(){
cin>>T;
while(T--){
cin>>n>>m>>k; res=0; g.clear(); vis[0]=flag=0;
for(int i=1;i<=n;i++) f[i]=i;
for(int i=1,a,b,c,d;i<=m;i++){
scanf("%lld %lld %lld %lld",&a,&b,&c,&d);
if(!d) res+=c,f[find(a)]=find(b);
else g.push_back({a,b,c}); vis[i]=0;
}
sort(g.begin(),g.end(),cmp);
for(int i=0;i<g.size()&&k;i++){
int x=find(g[i].u),y=find(g[i].v);
if(x!=y) k--,res+=g[i].w,f[x]=y,vis[i]=1;
}
for(int i=0;i<g.size()&&k;i++) if(!vis[i]) res+=g[i].w,k--;
for(int i=2;i<=n&&!flag;i++) if(find(i)!=find(1)) flag++;
printf("%lld\n",flag?-1:res);
}
return 0;
}