[bzoj2654][最小生成树][二分]tree

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

题解

神题呐不会啊只能膜题解了。。
如果直接对原树进行kruskal的话，求出来的白边可能＜need也可能＞need的对不
那么怎么人为控制白边在树里的数目同时保证黑边选择最小呐？
还是最小生成树，只不过我们二分一个值mid，每次check的时候每条白边加上这个mid值，这样可以保证白边在生成树里一定是单调不下降或者单调不上升的，而这个时候的黑边同样有保证一定是最小的。
跑kruskal的时候判一下白边的边数，如果出来的边数>=need就是正确状态，那么树里的白边就需要减小，这时候mid往大的二分
小于need的情况就往小的二分
答案继承的时候需要减去mid*need这么多，因为白边至少都多了这么多的权嘛
一个特别的地方：题里可能有权相等的黑白边，这样kruskal排序的时候黑白边顺序是不定的。那这样跑出来的边数就可能小于need而wa掉。所以我们排序的时候当边权相等的时候再按颜色排，白色在前

#include<cstdio>#include<cstring>#include<cstdlib>#include<algorithm>#include<cmath>using namespace std;struct node{    int x,y,c,op;}a[211000],e[211000];int fa[111000];int findfa(int x){    if(fa[x]!=x)fa[x]=findfa(fa[x]);    return fa[x];}int n,m,nd,cnt;bool cmp(node n1,node n2){    if(n1.c!=n2.c)return n1.c<n2.c;    else return n1.op<n2.op;}bool chk(int p){    for(int i=1;i<=m;i++)    {        e[i]=a[i];        if(a[i].op==0)e[i].c+=p;    }    for(int i=1;i<=n;i++)fa[i]=i;    sort(e+1,e+1+m,cmp);    int tmp=n,op=0;cnt=0;    for(int i=1;i<=m;i++)    {        int p=findfa(e[i].x),q=findfa(e[i].y);        if(p!=q)        {            fa[p]=q;            cnt+=e[i].c;tmp--;            if(e[i].op==0)op++;            if(tmp==1)break;        }    }    if(op>=nd)return true;    else return false;}int main(){    scanf("%d%d%d",&n,&m,&nd);    for(int i=1;i<=m;i++){scanf("%d%d%d%d",&a[i].x,&a[i].y,&a[i].c,&a[i].op);a[i].x++;a[i].y++;}    int l=-150,r=150,ans;    while(l<=r)    {        int mid=(l+r)/2;        if(chk(mid))        {            ans=cnt-nd*mid;            l=mid+1;        }        else r=mid-1;    }    printf("%d\n",ans);    return 0;}