SDOJ 3696 Tree

描述

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

输入

第一行V,E,need分别表示点数，边数和需要的白色边数。

接下来E行

每行s,t,c,col表示这边的端点(点从0开始标号)，边权，颜色(0白色1黑色)

输出

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

样例输入

2 2 1
0 1 1 1
0 1 2 0

样例输出

2

数据规模和约定

0%:V<=10

30%:V<=15

100%:V<=50000,E<=100000

所有数据边权为[1,100]中的正整数。

.......................

太困了！！！！！！

先贴个代码 明天再写思路

耶～

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<cmath>
 #define N 50010
 using namespace std;
 int n,m,k;
 int first[N],cnt;
 int a[N*],b[N*],c[N*],d[N*];
 int fa[N],sum;
 struct node
 {
     int u,v,w,nxt;
     int col;
 }e[N*];
 void ade(int x,int y,int z,int c,int i)
 {
     e[i].u=x;
     e[i].v=y;
     e[i].w=z;
     e[i].col=c;
 }
 bool cmp(const node &p,const node &q)
 {
     if(p.w!=q.w)
           return p.w<q.w;
     return p.col<q.col;
 }
 int la(int x)
 {
     if(fa[x]!=x) fa[x]=la(fa[x]);
     return fa[x];
 }
 bool kruskal(int x)
 {
     for(int i=;i<n;i++) fa[i]=i;
     for(int i=;i<=m;i++)
     {
         ade(a[i],b[i],c[i],d[i],i);
         if(d[i]==) e[i].w+=x;
     }
     sort(e+,e+m+,cmp);
     sum=;
     int num=,ans=;
     for(int i=;i<=m;++i)
     {
         if(la(e[i].u)!=la(e[i].v))
         {
             num++;
             sum+=e[i].w;
             fa[la(e[i].u)]=la(e[i].v);
             if(e[i].col==) ans++;
         }
         if(num==n-) break;
     }
     if(ans>=k)
         return true;
     return false;
 }
 int main()
 {
     scanf("%d%d%d",&n,&m,&k);
     for(int i=;i<=m;i++)
         scanf("%d%d%d%d",&a[i],&b[i],&c[i],&d[i]);
     int l=-,r=,mid;
     while(l<r)
     {
         mid=(l+r+)>>;
         if(kruskal(mid)) l=mid;
         else r=mid-;
     }
     kruskal(l);
     printf("%d",sum-k*l);
     return ;
 }