树，给你一个无向带权连通图，每条边是黑色或白色。C++代码实现

题目描述：

给你一个无向带权连通图，每条边是黑色或白色。

让你求一棵最小权的恰好有 n 条白色边的生成树。

题目保证有解。

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <stdlib.h>
using namespace std;
struct E
{
	int a,b,v,c,d;
}e[100050];
bool cmp(const E &x,const E &y)
{
	if(x.c==y.c) return x.d<y.d;
	return x.c<y.c;
}
int n,m,k,fa[50050],reimu;
int find(int x)
{
	return fa[x]==x?x:fa[x]=find(fa[x]);
}
void init(int x)
{
	for(int i=0;i<=n;i++)
		fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		if(e[i].d==0)
		{
			e[i].c=e[i].v+x;
		}
		else
			e[i].c=e[i].v;
	}
	sort(e+1,e+m+1,cmp);
}
int check(int mn)
{
	reimu=0;
	int re=0,nowe=0;
	init(mn);
	for(int i=1;i<=m;i++)
	{
		int x=e[i].a,y=e[i].b;
		int dx=find(x),dy=find(y);
		//printf("%d %d",x,y);
		if(dx!=dy)
		{
			reimu+=e[i].c;
			fa[dx]=dy;
			if(!e[i].d)
				re++;
			nowe++; 
			if(nowe==n-1)
			{
				break;
			}
		}
	}
	return re;
}
int main()
{
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d%d",&e[i].a,&e[i].b,&e[i].v,&e[i].d);
	}
	int l=-1000,r=1000,ans;
	while(l<r)
	{
		int mid=(l+r)>>1;
		int z=check(mid);
		if(z<k)r=mid;
		else l=mid+1,ans=reimu-k*mid;
	}
	printf("%d\n",ans);
}