最小生成树 tree(陈立杰)

问题 D: tree
时间限制: 3 Sec 内存限制: 512 MB
提交: 24 解决: 7
[提交][状态][讨论版]
题目描述
给你一个无向带权连通图，每条边是黑色或白色。让你求一棵最小权的恰好有need条白色边的生成树。
题目保证有解。
输入
第一行V,E,need分别表示点数，边数和需要的白色边数。
接下来E行,每行s,t,c,col表示这边的端点(点从0开始标号)，边权，颜色(0白色1黑色)。
输出
一行表示所求生成树的边权和。
V<=50000,E<=100000,所有数据边权为[1,100]中的正整数。
样例输入
2 2 1
0 1 1 1
0 1 2 0
样例输出
2
因为当前最小生成树上的白边是不确定的，所以通过修改白边的优先级来改变生成树上白边的数量。那么怎么修改呢？改边权。当然就想到二分答案了，给白边加上当前二分的值（-100,100）如果白边过多，加大值，过少反之。
注意，会有很多边边权一样，这时，白边数量q>k也可以是成立的（只要把一些等权的白边改成黑边就行了）注意最后答案要减去need*mid(二分的值)

     #include<cstdlib>#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;int n,m,k,mid,e,ans,q,adj[50005],f[50005];struct road{    int u,v,next,l,h;    bool friend operator <(road a,road b)    {        return a.l!=b.l ? a.l<b.l : a.h<b.h;    }}lu[100005];void add(int u,int v,int l,int h){lu[++e].u=u;lu[e].h=h;lu[e].v=v;lu[e].l=l;lu[e].next=adj[u];adj[u]=e;}inline int find(int x){return f[x]==x ? x:f[x]=find(f[x]);}inline void hb(int x,int y){int fx=find(x),fy=find(y);if(fx!=fy)f[fx]=fy;}inline int read(){    int sum=0,f=1;char x=getchar();    while(x<'0'||x>'9'){if(x=='-')f=-1;x=getchar();}    while(x>='0'&&x<='9'){sum=(sum<<1)+(sum<<3)+x-'0';x=getchar();}    return sum*f;}int check(){    ans=0;q=0;int w=0;    for(int i=1;i<=m;i++)if(!lu[i].h)lu[i].l+=mid;    sort(lu+1,lu+m+1);    for(int i=0;i<n;i++)f[i]=i;    for(int i=1;i<=m;i++)    {        if(find(lu[i].u)!=find(lu[i].v))        {            w++;            hb(lu[i].u,lu[i].v);            ans+=lu[i].l;            if(!lu[i].h)q+=1;        }        if(w==n-1)break;    }    for(int i=1;i<=m;i++)if(!lu[i].h)lu[i].l-=mid;    //cout<<q<<" "<<ans<<endl;    if(q>=k)return 1;    return 0;}int yjn(){//  freopen("nt2012_tree.in","r",stdin);//  freopen("nt2012_tree.out","w",stdout);    scanf("%d%d%d",&n,&m,&k);    int x,y,z,h;    for(int i=1;i<=m;i++)    {        x=read();y=read();z=read();h=read();        add(x,y,z,h);    }    int l=-100,r=100,w=0;    while(l<=r)    {        mid=(l+r)/2;        if(check())      {           l=mid+1,w=ans-k*mid;       }       else r=mid-1;   }    cout<<w;}int qty=yjn();int main(){;}