[bzoj2654] tree

  一开始以为先取need条最短的白边就行了。。然而那样子的话可能图根本没法联通= =

  网上题解讲的挺清晰的。。就是二分把全部白边加上mid,然后看mst里面有多少条白边。有need条白边的时候再把加上的值减去,就是答案了。

  但可能出现取不了need条白边的情况(二分mid取到>need条,二分mid+1时取到<need条)

  这时候就是因为图中有一些黑边和白边权值相同。。那显然我们取的是黑边还是白边是可以替换的。。

  所以取到>=need条白边的时候就可以更新答案。注意权值相同的话优先先取白边

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<cstring>
 4 #include<algorithm>
 5 using namespace std;
 6 const int maxn=100233;
 7 struct zs{
 8     int u,v,dis;
 9 }e[maxn],e1[maxn];int t,t1;
10 int fa[maxn],id[maxn];
11 int i,j,k,n,m,ans,a,b,c,d,now,l,r,mid;
12 
13 int ra;char rx;
14 inline int read(){
15     rx=getchar(),ra=0;
16     while(rx<'0'||rx>'9')rx=getchar();
17     while(rx>='0'&&rx<='9')ra*=10,ra+=rx-48,rx=getchar();return ra;
18 }
19 bool cmp(int a,int b){return e[a].dis==e[b].dis?a<b:e[a].dis<e[b].dis;}
20 inline int getfa(int x){return fa[x]!=x?fa[x]=getfa(fa[x]):x;}
21 inline int get(int x){
22     register int i,sm=1,num0=0;now=0;
23     for(i=1;i<=t;i++)e[i].dis+=x;
24     for(i=1;i<=n;i++)fa[i]=i;
25     sort(id+1,id+1+m,cmp);
26 //    for(i=1;i<=m;i++)printf("! %d   %d--%d %d    %d %d\n",id[i],e[id[i]].u,e[id[i]].v,e[id[i]].dis,getfa(e[id[i]].u),getfa(e[id[i]].v));
27     for(i=1;i<=m&&sm<n;i++)
28         if((a=getfa(e[id[i]].u))!=(b=getfa(e[id[i]].v)))
29             fa[a]=b,sm++,
30             num0+=id[i]<=t,
31             now+=e[id[i]].dis;//,printf("  %d ",id[i]);
32     
33     for(i=1;i<=t;i++)e[i].dis-=x;
34 //    printf("   x:%d num0:%d now:%d   sm:%d\n",x,num0,now,sm);
35     return num0;
36 }
37 int main(){register int i;
38     n=read(),m=read(),k=read();
39     for(i=1;i<=m;i++){
40         a=read()+1,b=read()+1,c=read(),d=read();
41         if(d)e1[++t1]=(zs){a,b,c};
42         else e[++t]=(zs){a,b,c};
43     }
44     for(i=t+1;i<=m;i++)e[i]=e1[i-t];
45     for(i=1;i<=m;i++)id[i]=i;
46     l=-101,r=101;ans=1002333333;
47     while(l<=r){
48         mid=(l+r)/2;//printf("    %d -- %d\n",l,r);
49         if(get(mid)>=k)ans=now-mid*k,l=mid+1;
50             else r=mid-1;
51     }
52     printf("%d\n",ans);
53     return 0;
54 }
View Code

 

posted @ 2016-03-21 13:45  czllgzmzl  阅读(298)  评论(0编辑  收藏  举报