CF958E2 Guard Duty (medium)

传送门

一道小清新题

首先转化一下问题,将题目给出的\(t\)数组\(sort\)之后差分一下。

问题就变成了给出一个长度为\(n-1\)的序列,选出\(k\)个不相邻的数使得和最小

很容易发现这是一个下凸的函数,\(\rm wqs\)二分去掉\(k\)的限制之后那个\(dp\)就很简单了

\(f[i][0/1]\)表示前\(i\)个数,第\(i\)个数不选/选的最小和

代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
#define rg register
#define int long long
void read(int &x){
    char ch;bool ok;
    for(ok=0,ch=getchar();!isdigit(ch);ch=getchar())if(ch=='-')ok=1;
    for(x=0;isdigit(ch);x=x*10+ch-'0',ch=getchar());if(ok)x=-x;
}
const int maxn=5e5+10,inf=1e9;
int sum,k,n,a[maxn],b[maxn],ans;
struct oo{int x,y;}f[maxn][2];
bool operator<(oo a,oo b){return a.x==b.x?a.y<b.y:a.x<b.x;}
bool check(int x){
    f[1][0]=(oo){0,0},f[1][1]=(oo){b[1]-x,1};
    for(rg int i=2;i<n;i++){
        f[i][0]=min(f[i-1][0],f[i-1][1]);
        f[i][1]=(oo){f[i-1][0].x+b[i]-x,f[i-1][0].y+1};
    }
    oo now=min(f[n-1][0],f[n-1][1]);
    return sum=now.x,now.y<=k;
}
signed main(){
    read(k),read(n);
    for(rg int i=1;i<=n;i++)read(a[i]);
    sort(a+1,a+n+1);
    for(rg int i=1;i<n;i++)b[i]=a[i+1]-a[i];
    int l=-inf,r=inf;
    while(l<=r){
        int mid=(l+r)>>1;
        if(check(mid))l=mid+1,ans=mid*k+sum;
        else r=mid-1;
    }
    printf("%lld\n",ans);
}
posted @ 2019-07-13 15:33 蒟蒻--lichenxi 阅读(...) 评论(...) 编辑 收藏