bzoj4582 [Usaco2016 Open]Diamond Collector（dp）

县用一个指针处理出dp[i],表示以a[i]为最小值最长往后延伸距离,然后后缀最大值优化一下即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 50010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,K,a[N],f[N],dp[N];//dp[i],以a[i]为最小值最长往后延伸距离,f[i]后缀最大值
int main(){
//  freopen("a.in","r",stdin);
    n=read();K=read();
    for(int i=1;i<=n;++i) a[i]=read();
    sort(a+1,a+n+1);int r=1;
    for(int i=1;i<=n;++i){
        while(r<=n&&a[r]<=a[i]+K) ++r;
        dp[i]=r-i;
    }for(int i=n;i>=1;--i) f[i]=max(f[i+1],dp[i]);
    int ans=0;
    for(int i=1;i<=n;++i)
        ans=max(ans,dp[i]+f[i+dp[i]]);
    printf("%d\n",ans);
    return 0;
}