【bzoj 2654】 tree

Description

给你一个无向带权连通图，每条边是黑色或白色。让你求一棵最小权的恰好有need条白色边的生成树。

题目保证有解。

 

Input

第一行V,E,need分别表示点数，边数和需要的白色边数。

接下来E行,每行s,t,c,col表示这边的端点(点从0开始标号)，边权，颜色(0白色1黑色)。

 

Output

一行表示所求生成树的边权和。

V<=50000,E<=100000,所有数据边权为[1,100]中的正整数。

 

 

Sample Input

2 2 1
0 1 1 1
0 1 2 0

Sample Output

2

HINT

 

原数据出错，现已更新 by liutian,但未重测---2016.6.24

对于这道题，考虑kruskal算法的过程，我们可以通过给白边上一个权值k来改变白边在排序后的位置，显然若则加入的白边数然后二分k即可，下面是程序：

#include<stdio.h>
#include<algorithm>
#include<iostream>
using namespace std;
const int N=50005,M=100005;
struct edge{
	int u,v,w;
	bool co;
}e[M];
int n,m,need,ans=0x7fffffff,k,f[N];
bool cmp(const edge &a,const edge &b){
	return a.w+a.co*k<b.w+b.co*k;
}
int findf(int u){
	return f[u]=u==f[u]?u:findf(f[u]);
}
bool check(){
	int i,x,y,cnt=0,sum=0;
	sort(e+1,e+m+1,cmp);
	for(i=0;i<n;i++){
		f[i]=i;
	}
	for(i=1;i<=m;i++){
		x=findf(e[i].u),y=findf(e[i].v);
		if(x!=y){
			f[x]=y;
			cnt+=e[i].co;
			sum+=e[i].w;
		}
	}
	if(cnt>=need){
		ans=min(ans,sum);
	}
	return cnt>need;
}
int main(){
	int l,r,i;
	scanf("%d%d%d",&n,&m,&need);
	for(i=1;i<=m;i++){
		scanf("%d%d%d%d",&e[i].u,&e[i].v,&e[i].w,&l);
		e[i].co=!l;
	}
	l=-101,r=101;
	while(l<=r){
		k=l+r>>1;
		if(check()){
			l=k+1;
		}
		else{
			r=k-1;
		}
	}
	printf("%d\n",ans);
	return 0;
}