【BZOJ】2654 tree 二分+kruskal

题目传送门

好神的想法啊……这个二分好迷啊……

我们可以二分一个值$mid$，把所有白色的边都加上这个值，在最后计算边权和时减回去，也就相当于改变的白色边的优先级。

然后刷kruskal，如果mst中的白色边数不小于给定的need则验证成功，修改二分的边界继续二分。

附上AC代码：

#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;

const int N=5e4+10;
struct note{
    int x,y,w,c;
    bool operator < (const note lyf) const {return w==lyf.w?c<lyf.c:w<lyf.w;}
}a[N<<1],s[N<<1];
int n,m,k,sum,cnt,f[N],l,r,mid,ans;

inline char nc(void){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
    return (void)(a*=f);
}

inline int gf(int x){return x==f[x]?x:f[x]=gf(f[x]);}
inline bool check(int mid){
    sum=cnt=0;
    for (int i=1; i<=n; ++i) f[i]=i;
    for (int i=1; i<=m; ++i) s[i]=a[i],s[i].c==0?s[i].w+=mid:0;
    sort(s+1,s+1+m);
    for (int i=1; i<=m; ++i){
        int fx=gf(s[i].x),fy=gf(s[i].y);
        if (fx!=fy){
            f[fx]=fy,sum+=s[i].w;
            if (!s[i].c) ++cnt;
        }
    }
    return cnt>=k;
}

int main(void){
    read(n),read(m),read(k);
    for (int i=1; i<=m; ++i) read(a[i].x),read(a[i].y),read(a[i].w),read(a[i].c),++a[i].x,++a[i].y;
    l=-105,r=105;
    while (l<r)
        if (check(mid=l+r>>1)) l=mid+1,ans=sum-k*mid;
        else r=mid-1;
    printf("%d\n",ans);
    return 0;
}