2019ICPC南昌区域赛 Bob‘s Problem 题解

题目
https://nanti.jisuanke.com/t/42580
思路：
首先，黑边是一定要全部选中的。
但是为了后面使用生成树，在加入黑边的时候，只对不存在并查集内的边计算入生成的树的边数
加入白边时，考虑的因素有：
k于n-1的关系。

1. k条白边加入后，生成树的边数 等于 n-1 ，则 恰好满足题目要求，答案为这k条白边 + 所有黑白的权值。
2. 生成树的边数 小于 n-1，则满足不了题目要求，答案为-1
3. 生成树的边数 大于 n-1，则除了树的边以外，继续从大到小添加白边，直到k条白边（多多益善）。

后面贪心从权值大的白边开始，为优先保证所有点连通，先生成树， 如果两点不在并查集内就加入。

• 当加入的白边数等于k时，一定要退出。
  退出的值取决于，是否生成了树，即 边数是否等于n-1（n为点数）。
  生成了树就表示任意两点相连，返回答案。
  没有，则该例子没有答案，返回-1.

• 当已经生成了树后，白边数 < k，则在剩下的白边中，按权值从大到小将其加入。

#include<iostream>
#include<vector>
#include<algorithm>
#include<cstring>
#include<cstdio>
#define int long long
using namespace std;
const int maxn = 1e6+7;
struct edge{
	int u;
	int v;
	int w;
	int color;
};
vector<edge> e;
int n,m,k;
int f[maxn];
long long ans = 0;
long long cnt = 0;
bool vis[maxn];

bool cmp(edge a, edge b){
	return a.w > b.w;
}
int find(int x){
	if(f[x] != x){
		f[x] = find(f[x]);
	}
	return f[x];
}
int kurskal(){
	int k_count = 0;
	for(int i = 0;i < e.size();i++){
		edge newE = e[i];
		if(newE.color == 1){
			int x = find(newE.u);
			int y = find(newE.v);
			if(x!=y){
				f[x] = y;
				ans+=newE.w;
				vis[i]=1;
	//			cout<<"ans = "<<ans<<endl;
				cnt++;
				k_count++;
				
				// k 满足条件了 一定得返回
				// 已经生成树了，说明相连了，跳出，选最大的边 
				if(k_count >= k){
					if(cnt == n-1){
	//					cout<<"k ans = "<<ans<<endl;
						return ans;
					}else{
						return -1; 
					}
				}

			}			
		}

	}
	
	
	for(int i=0;i<e.size();i++){
		edge newE = e[i];
		if(newE.color == 1&&vis[i] == 0 && k_count < k){
			ans+=newE.w;
//			cout<<"ans = "<<ans<<endl;
			k_count++;
		}
	}
	
	if(cnt == n-1){
		return ans;
	}else{
		return -1; 
	}
}
void read(){
	e.clear();
	n=m= k = 0;
	cin>>n>>m>>k;
	ans = 0;
	cnt = 0;
	
	for(int i = 0;i <= n;i++){
		f[i] = i;
		vis[i]=0;
	}
	
	for(int i = 0;i < m;i++){
		edge newE;
		scanf("%lld%lld%lld%lld",&newE.u,&newE.v,&newE.w,&newE.color);
//		printf("%d%d%d%d\n\n",newE.u,newE.v,newE.w,newE.color);
		e.push_back(newE);
		

	}
	//对于白边也是先选掉 
	sort(e.begin(),e.end(),cmp);
	
	for(int i = 0;i<e.size();i++){
		edge newE = e[i];
		
		if(newE.color == 0){
			int x = find(newE.u);
			int y = find(newE.v);
			
			ans+=newE.w;
			if(x!=y){
//				cout<<"enter"<<endl;
				f[x] = y;
//				cout<<"ans = "<<ans<<endl;
				cnt++;
			}
	    }
	}
	long long now_ans = kurskal();
	printf("%lld\n",now_ans);
	

}
signed main()
{
	int t;
	cin>>t;
	while(t--){
		read();
	}
	return 0;
}