BZOJ2654 tree 二分+最小生成树

给出$N\le 5e4,M\le 1e5$的无向连通图，一些边是白色，一些边是黑色的。然后求问满足恰好的$K$条边是白色的最小生成树的值是多少。
是有一个比较新奇的思路，有博主把这个称为凸优化，具体做法是先给所有的白边都增加一个相同的权值，然后对这个新图求最小生成树。那么如果满足树上的白边恰好是$K$条，那么就更新答案。
由于显然增加的权值越大，白边的数量就会越少。所以可以二分增加的权值。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=5e4+7;
const int M=1e5+7;
struct edge {
	int u,v,w,col;
	bool operator <(const edge &rhs) const {
		if(w<rhs.w) return 1;
		else if(w==rhs.w) return col<rhs.col;
		else return 0;
	}
}e[M];
int fa[N]; 
int n,m,nd;
int find(int x) {
	if(fa[x]==x) return fa[x];
	else return fa[x]=find(fa[x]);
}
pair<int,int> check(int x) {
	for(int i=1;i<=m;i++) if(!e[i].col) e[i].w+=x;
	sort(e+1,e+1+m);
	for(int i=1;i<=n;i++) fa[i]=i;
	int tot=0,white=0,ans=0;
	for(int i=1;i<=m;i++) {
		int U=find(e[i].u);
		int V=find(e[i].v);
		if(U!=V) {
			fa[U]=V;
			ans+=e[i].w;
			if(!e[i].col) white++;
			tot++;
		} 
		if(tot==n-1) break;
	}
	for(int i=1;i<=m;i++) if(!e[i].col) e[i].w-=x;
	return make_pair(white,ans);
}
int main() {
	scanf("%d%d%d",&n,&m,&nd);
	for(int i=1;i<=m;i++) {
		int u,v,w,col;
		scanf("%d%d%d%d",&u,&v,&w,&col);
		u++,v++; 
		e[i].u=u;e[i].v=v;
		e[i].w=w;e[i].col=col;
	}
	int l=-100,r=100,ans=-1;
	while(l<=r) {
		int mid=(l+r)>>1; 
		pair<int,int> result=check(mid);
		if(result.first>=nd) l=mid+1,ans=result.second-result.first*mid;
		else r=mid-1;
	}
	printf("%d\n",ans);
	return 0;
}