「BZOJ2654」tree

「BZOJ2654」tree

最小生成树+二分答案。

最开始并没有觉得可以二分答案，因为答案并不单调啊。

其实根据题意，白边的数目肯定大于need条，而最小生成树的白边数并不等于need（废话），可以二分将每条白边的权值+mid，这样就可以控制最小生成树中白边的条数，

对于一个mid，将所有的白边权值加mid，然后跑kruskal，求出最小生成树中白边的个数num以及此时的权值和ans（要减去mid*need），如果num=need直接输出ans，如果num<need

则让r=mid继续二分，如果num>need,则要记录一下此时的答案（注意此时的答案并不是ans-mid*num，而是ans-mid*need），因为有可能会出现如下情况：要求8条白边，mid=0时num=12，mid=1时num=0，无法是num恰好等于need，此时记录的数据就是答案。

#include<algorithm>
#include<iostream>
#include<cstdio>
#define INF 0x7fffffff
using namespace std;
struct edge
{
	int u,v,w,c,nxt;
	#define u(x) ed[x].u
	#define v(x) ed[x].v
	#define w(x) ed[x].w
	#define c(x) ed[x].c
	#define n(x) ed[x].nxt
	friend bool operator < (edge a,edge b)
	{return a.w==b.w?(a.c<b.c):(a.w<b.w);}
}ed[100010];
int first[50010],num_e;
#define f(x) first[x]
int f[50010];
int v,e,n,nb;
int l=-500,r=500,mid,ans,eans=INF;

int getf(int x){return (f[x]==x)?x:f[x]=getf(f[x]);}
void hb(int x,int y){x=getf(x),y=getf(y),f[y]=x;}
inline void add(int u,int v,int w,int c);
int kruskal()
{
	int num=0;ans=0;
	for(int i=0;i<=v;i++)f[i]=i;
	sort(ed+1,ed+num_e+1);
	for(int i=1;i<=num_e;i++)
	if(getf(u(i))!=getf(v(i)))
	{
		ans+=w(i);
		hb(u(i),v(i));
		if(!c(i))num++;
	}
	return num;
}
signed main()
{
//	freopen("9.in","r",stdin);
//	freopen("in.txt","r",stdin);

	scanf("%d%d%d",&v,&e,&n);
	for(int i=0;i<=v;i++)f[i]=i;
	int s,t,w,c;
	for(int i=1;i<=e;i++)
	{
		scanf("%d%d%d%d",&s,&t,&w,&c);
		add(s,t,w,c);
		if(!c)nb++;
	}
	sort(ed+1,ed+num_e+1);
	int num;
	while(l<r-1)
	{
		mid=(l+r)>>1;
		for(int i=1;i<=num_e;i++)if(!c(i))w(i)+=mid;
		num=kruskal();
		if(num==n){printf("%d\n",ans-mid*num);return 0;}
		else if(num<n)r=mid;
		else {l=mid;eans=ans-mid*n;}
		for(int i=1;i<=num_e;i++)if(!c(i))w(i)-=mid;
	}
	printf("%d\n",eans);
}
inline void add(int u,int v,int w,int c)
{
	++num_e;
	u(num_e)=u;
	v(num_e)=v;
	w(num_e)=w;
	c(num_e)=c;
	n(num_e)=f(u);
	f(u)=num_e;
}