BZOJ2654 tree （wqs二分）

2022-04-05 10:04:47 阅读：144 来源： 互联网

标签：use BZOJ2654 wqs int tree 白边 mid 权值 need

题目描述

给你一个无向带权连通图，每条边是黑色或白色。让你求一棵最小权的恰好有need条白色边的生成树。题目保证有解。一个最小生成树问题，但是我们要选need条白边，我们用g(i)表示选取i条白边的最优方案（生成树的权值最小），那么可以大致猜出g(i)是关于i的一个下凸函数，可以发现斜率k是有单调性的，我们二分这个斜率k，相当于给每条白边的权值加上一个k，统计数量use，如果use>=need，说明k小了，要增大，同理，use<need，要减小k。

那么问题来了，如果说当前白边加上mid后，白边条数use>need了，如果加上mid+1后，use<need了要怎么办？

题目中说到了：保证有解，所以出现上述情况时一定有黑边==白边的边权

所以我们只需要把一条黑边换成白边就好，即我们排序时如果黑边权值等于白边，则白边优先。

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 #define N 50001
 4 #define M 100001
 5 int n,m,k;
 6 struct node{
 7     int u,v,w,c;
 8 }e[M];
 9 int use,sum,fa[N];
10 bool cmp(node a,node b){
11     if(a.w!=b.w) return a.w<b.w;
12     return a.c<b.c;//黑边白边权值一样，优先选白边 
13 }
14 int find(int x){
15     return fa[x]==x?x:fa[x]=find(fa[x]);
16 } 
17 void check(int x){
18     for(int i=1;i<=m;i++)
19         if(e[i].c==0) e[i].w+=x;//白边加权值 
20     sort(e+1,e+m+1,cmp);
21     for(int i=1;i<=n;i++) fa[i]=i;
22     int now=0,fu,fv,j=1;
23     use=0,sum=0;
24     while(now!=n-1){
25         fu=find(e[j].u);fv=find(e[j].v);
26         if(fu!=fv){
27             now++;
28             fa[fu]=fv;
29             if(e[j].c==0) ++use;//统计使用的白边数量 
30             sum+=e[j].w;
31         }
32         ++j;
33     }
34     for(int i=1;i<=m;i++)
35         if(e[i].c==0) e[i].w-=x;//还原 
36 }
37 
38 int main()
39 {
40     scanf("%d%d%d",&n,&m,&k);
41     for(int i=1;i<=m;i++){
42         scanf("%d%d%d%d",&e[i].u,&e[i].v,&e[i].w,&e[i].c);
43         ++e[i].u;++e[i].v;//题目是编号从0开始，要+1 
44     }
45     int l=-101,r=101,mid,ans;//值域[1,100]； 
46     while(l<=r){
47         mid=l+r>>1;check(mid);
48         if(use>=k){
49             ans=sum-k*mid;
50             l=mid+1;
51         }
52         else r=mid-1;
53     }
54     cout<<ans;
55     return 0;
56 }

标签：use,BZOJ2654,wqs,int,tree,白边,mid,权值,need
来源： https://www.cnblogs.com/yhxnoerror/p/16101502.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

BZOJ2654 tree （wqs二分）

题目描述