p2619 [国家集训队2]Tree I [wqs二分学习]

分析

https://www.cnblogs.com/CreeperLKF/p/9045491.html

反正这个博客看起来很nb就对了

但是不知道他在说啥

实际上wqs二分就是原来的值dp[x]表示选x个的最优解满足是一个凸包

然后就可以二分一个数

让有限制的一类数全部给权值加上这个二分的数

然后判断即可

这个题就是典型的模板题

代码

#include<bits/stdc++.h>
using namespace std;
const int inf = 1e9+;
int n,m,sum,res,cnt,tot,fa[];
struct node {
    int x,y,z,col;
};
node d[],d1[];
inline int sf(int x){return fa[x]==x?x:fa[x]=sf(fa[x]);}
inline bool cmp(const node x,const node y){return x.z==y.z?x.col<y.col:x.z<y.z;}
inline int ck(int mid){
    int i,j,k;
    for(i=;i<=m;i++)d[i]=d1[i];
    for(i=;i<=m;i++)if(!d[i].col)d[i].z+=mid;
    res=tot=cnt=;
    for(i=;i<=n;i++)fa[i]=i;
    sort(d+,d+m+,cmp);
    for(i=;i<=m;i++){
      int x=d[i].x,y=d[i].y;
      if(sf(x)!=sf(y)){
          fa[sf(x)]=sf(y);
          cnt++;
          res+=d[i].z;
          if(!d[i].col)tot++;
      }
      if(cnt==n-)break;
    }
    return tot;
}
int main(){
    int i,j,k,le=-,ri=;
    scanf("%d%d%d",&n,&m,&sum);
    for(i=;i<=m;i++)scanf("%d%d%d%d",&d1[i].x,&d1[i].y,&d1[i].z,&d1[i].col),d1[i].x++,d1[i].y++;
    while(ri-le>){
      int mid=(le+ri)>>;
      if(ck(mid)>=sum)le=mid;
        else ri=mid;
    }
    ck(le);
    printf("%d\n",res-sum*le);
    return ;
}