BZOJ 2654: tree kruskal 二分

2654: tree

Time Limit: 30 Sec  Memory Limit: 512 MB
Submit: 2178  Solved: 893
[Submit][Status][Discuss]

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

懒懒懒

犯了一堆SB错误 看来晨起刷水掉智商啊

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

inline 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<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)x=-x,putchar('-');if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=50010;

int n,m,need,ans;

struct EDGE{int fr,to,val;bool col;friend bool operator <(const EDGE &x,const EDGE &y){return x.val==y.val?x.col<y.col:x.val<y.val;}}e[N<<1];

int fa[N];

inline int find(int x)
{
	int t=x;
	while(t^fa[t])t=fa[t];
	while(x^fa[x])x=fa[x],fa[x]=t;
	return t;
}

bool check(int val)
{
	register int i,x,y,u,v,now=0,hv=0;
	ans=0;
	for(i=1;i<=n;++i)fa[i]=i;
	for(i=1;i<=m;++i)if(!e[i].col)e[i].val+=val;
	sort(e+1,e+1+m);
	for(i=1;i<=m;++i)
	{
		u=e[i].fr,v=e[i].to;
		x=find(u);y=find(v);
		if(x^y)
		{
			fa[x]=y;
			now++;ans+=e[i].val;
			if(!e[i].col)hv++;
		}
		if(now==n-1)break;
	}
	for(i=1;i<=m;++i)if(!e[i].col)e[i].val-=val;
	return hv>=need;
}

int main()
{
	n=read();m=read();need=read();
	register int i,l=-105,r=105,mid,res;
	for(i=1;i<=m;++i){e[i].fr=read()+1;e[i].to=read()+1;e[i].val=read();e[i].col=read();}
	while(l<=r)
	{
		mid=(l+r)>>1;
		if(check(mid))res=ans-mid*need,l=mid+1;
		else r=mid-1;
	}
	print(res);puts("");
	return 0;
}
/*
2 2 1
0 1 1 1
0 1 2 0

2
*/