BZOJ 2654 tree（二分答案+并查集）

【题目链接】 http://www.lydsy.com/JudgeOnline/problem.php?id=2654

【题目大意】

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

【题解】

　　我们发现对于选中的边白色是从小到大的，黑色也是从小到大的，
　　因此我们对所有的白色边加一个权值，那么排序后做mst选取的白色边数量增减性单调，
　　对于增加的权值进行二分，验证能否满足要求即可。

【代码】

#include <cstdio>
#include <algorithm>
using namespace std;
const int N=100010;
struct data{int a,b,u,c;}p
;
bool cmp(data a,data b){return a.u<b.u;}
int ans,res,v,e,need,f
;
int sf(int x){return f[x]==x?x:f[x]=sf(f[x]);}
bool check(int x){
int cnt=0; res=0;
for(int i=0;i<v;i++)f[i]=i;
for(int i=1;i<=e;i++){if(!p[i].c)p[i].u+=x;}
sort(p+1,p+e+1,cmp);
for(int i=1;i<=e;i++){if(!p[i].c)p[i].u-=x;}
for(int i=1;i<=e;i++){
if(p[i].u<x)continue;
if(sf(p[i].a)!=sf(p[i].b)){
res+=p[i].u;
f[sf(p[i].a)]=sf(p[i].b);
cnt+=(p[i].c==0);
}
}return cnt>=need;
}
int main(){
while(~scanf("%d%d%d",&v,&e,&need)){
for(int i=1;i<=e;i++){scanf("%d%d%d%d",&p[i].a,&p[i].b,&p[i].u,&p[i].c);}
int l=-100,r=100;
while(l<=r){
int mid=(l+r)>>1;
if(check(mid))l=mid+1,ans=res;
else r=mid-1;
}printf("%d\n",ans);
}return 0;
}