tree陈立杰

简单的kruskal不能保证白边个数，那么我们队百变统一加上一个权值，这样来控制白边的个数，也并不改变白边内部相对关系，二分判断加入的权值，如果num==need，用此时的sum-need*x(二分的权值)

也有可能出现一种情况，mid时白边个数太多，mid+1时白边个数太少，这样是因为mid时白边黑边权值相同的太多，这时我们只需要按颜色为第二关键字排序，保证相同权值下白边先被选中，二分时对于num>=need的更新答案

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 100005
using namespace std;
int n,m,need;
int sum,ans;
struct edge
{
    int s,t,w,col;
}b[maxn],a[maxn];
int fa[maxn];
inline int read()
{
       int x=0;char ch=getchar();
       while(ch<'0'||ch>'9') ch=getchar();
       while(ch>='0'&&ch<='9') {x=x*10+ch-'0';ch=getchar();}
       return x;
}
int cmp(const edge &a,const edge &b)
{
    return a.w==b.w ? a.col<b.col : a.w<b.w; 
}
int find(int x)
{
    if(fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}
int check(int x)
{
    // printf("%d\n",x);
     memcpy(a,b,sizeof(b));
     for(int i=0;i<=n;i++) fa[i]=i;
     for(int i=1;i<=m;i++)
     if(a[i].col==0) a[i].w+=x;
     sort(a+1,a+m+1,cmp);
     sum=0;
     int white_num=0,num=0;
     for(int i=1;i<=m;i++)
     {
         int fx=find(a[i].s);
         int fy=find(a[i].t);    
         if(fx!=fy){
             fa[fx]=fy; num++;
             sum+=a[i].w;
             if(a[i].col==0)  white_num++;
         }
         if(num==n-1) break;
     }
     if(white_num==need) return 1;
     if(white_num>need) return 2;
     else return 0;
}
void erfen(int l,int r)
{
     ans=0x7fffffff; 
     while(l<=r){
         int mid=(l+r)/2;
         int t=check(mid);
         if(t==1){ ans=sum-mid*need; break; }
         if(t==2){ ans=sum-mid*need; l=mid+1; }
         else  r=mid-1;
     }
}
int main()
{
    //freopen("in.txt","r",stdin);
    freopen("nt2012_tree.in","r",stdin);
    freopen("nt2012_tree.out","w",stdout);
    n=read();m=read();need=read();
    for(int i=1;i<=m;i++){
       b[i].s=read();b[i].t=read();b[i].w=read();b[i].col=read();
    }
    erfen(-100,100);
    printf("%d\n",ans);
    //while(1);
    return 0;
}