2019ICPC南昌 Bob's Problem

题目链接：Bob’s Problem

题意：就是给你一些白边和一些黑边，让你随便选，但是白边不超过k条。

很明显黑边能选就选，然后白边贪心从大往小，先从大到小让整个图连通，然后从大到小选没选过的即可。

---

AC代码：

#pragma GCC optimize(2)
#include
#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;
}