BZOJ 2654 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

---

Solution

刚开始看到题目一脸懵，恰好need条白边，这怎么找啊（无数贪心飘过~）
于是去%大佬，get了正解。

---

我们可以二分一个mid，使每条白边加上mid，然后排序做最小生成树。
这样可以调整黑白边的顺序，使最小生成树中的白边数改变。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;

#define N 50050
#define E 100100

struct edge
{
    int x,y,l,t;
};

edge side[E];
int fa[N];
int n,m,need,tot,ans,an,mid,l,r;

bool comp(edge x,edge y)
{
    if (x.l<y.l) return true;
    if (x.l==y.l && x.t<y.t) return true;
    return false;
}

int find(int x)
{
    if (fa[x]==x) return x;
    fa[x]=find(fa[x]);
    return fa[x];
}

int mst()
{
    for (int i=1;i<=m;i++)
      if (side[i].t==0) side[i].l+=mid;
    for (int i=0;i<=n;i++)
      fa[i]=i;
    ans=0;
    sort(side+1,side+m+1,comp);
    int x,y,s=0;
    for (int i=1;i<=m;i++)
    {
        x=find(side[i].x);
        y=find(side[i].y);
        if (x==y) continue;
        fa[x]=y;
        if (side[i].t==0) s++;
        ans+=side[i].l;
    }
    for (int i=1;i<=m;i++)
      if (side[i].t==0) side[i].l-=mid;
    return s;
}

int main()
{
    scanf("%d%d%d",&n,&m,&need);
    for (int i=1;i<=m;i++)
      scanf("%d%d%d%d",&side[i].x,&side[i].y,&side[i].l,&side[i].t);
    l=-100,r=100;
    while (l<=r)
    {
        mid=(l+r)/2;
        int s=mst();
        //printf("%d %d %d %d\n",l,r,mid,s);
        if (s<need) r=mid-1;
          else 
            {
                l=mid+1;
                an=ans-need*mid;
            }
    }
    printf("%d\n",an);
    return 0;
}