【题解】sdoj3696tree（同bzoj2654）（2018-08-07集训T3）生成树问题+二分

题目描述

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

输入

第一行V,E,need分别表示点数，边数和需要的白色边数。
接下来E行
每行s,t,c,col表示这边的端点(点从0开始标号)，边权，颜色(0白色1黑色)

输出

一行表示所求生成树的边权和。

样例输入

2 2 1
0 1 1 1
0 1 2 0

样例输出

2

数据规模和约定

10%:V<=10
30%:V<=15
100%:V<=50000,E<=100000
所有数据边权为[1,100]中的正整数。

---

题解

考虑如何才能让白边显得更（不）重要，即在每条白边上（加上）减去一个值。
我们可以二分这个值，然后用寻常方法做最小生成树。 统计在此最小生成树里有多少白边。
然后我们就可以找到一个合适的值，带这个权做一次最小生成树。
在计算答案的时候把这些值补偿回去就做完了。

#include<cstdio>
#include<vector>
#include<algorithm>
#define PB(v) push_back(v)
using namespace std;
const int N=5e4+10;
const int M=1e5+10;
int n,m,k;
struct Edge{
    int u,v,prew,noww,color;//端点,最初边权，修改后的边权，边的颜色 
    bool operator <(const Edge rhs)const{
    return noww<rhs.noww||(noww==rhs.noww&&color<rhs.color);}
};
vector<Edge>w;//白边 
vector<Edge>b;//黑边 
int set[N];
int sum,cnt;
int findset(int x)
{
    return set[x]==x?x:set[x]=findset(set[x]);
}
bool check(int x)
{
    for(int i=0;i<w.size();i++)
    w[i].noww=w[i].prew+x;//修改白边边权 
    Edge edge[M];
    merge(w.begin(),w.end(),b.begin(),b.end(),edge);//修改后白边和黑边按顺序加入容器中
    for(int i=0;i<n;i++) 
    set[i]=i;
    sum=cnt=0;
    for(int i=0;i<m;i++)//跑最小生成树 
    {
        int fu=findset(edge[i].u);
        int fv=findset(edge[i].v);
        if(fu==fv)continue;
        set[fu]=fv;
        sum+=edge[i].noww;
        if(!edge[i].color)cnt++;
    }
    return cnt>=k;//白边数目与需要数目 
}
int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d%d%d",&n,&m,&k);
    for(int i=0;i<m;i++)
    {
        Edge e;
        scanf("%d%d%d%d",&e.u,&e.v,&e.prew,&e.color);
        e.noww=e.prew;
        if(e.color==0)w.PB(e);
        else b.PB(e);
    }
    sort(b.begin(),b.end());
    sort(w.begin(),w.end());
    int l=-101,r=101;
    while(l<r-1)//二分修改的边权 
    {
        int mid=l+r>>1;
        if(check(mid))l=mid;//白边多了说明加的数过小 
        else r=mid;//白边少了说明加的数过大 
    }
    check(l);
    printf("%d\n",sum-l*k);
    return 0;
}