题意
在数轴上有 n n n 块宝石,当你走到一个点时,可以获得点上所有的宝石。
若前一步走了 c c c 个单位长度,那么下一步可以走 c − 1 , c , c + 1 c-1,c,c+1 c−1,c,c+1 个单位长度。
你最开始在原点,可以向右走 d d d 个单位长度,求最多可获得多少宝石。
分析
设 f i , j f_{i,j} fi,j 表示在第 i i i 个点,上一步走 j j j 个单位长度可获得的最大宝石数。
但是 n , p i n,p_i n,pi 的范围是 3 × 1 0 4 3\times 10^4 3×104,二维数组会 MLE,考虑怎么优化。
因为最远的宝石在 3 × 1 0 4 3\times 10^4 3×104,设步数变化 x x x 次,则最大的 x x x 满足 ( 2 d + x ) ( x + 1 ) 2 ≤ 3 × 1 0 4 \frac{(2d+x)(x+1)}{2}\le 3\times 10^4 2(2d+x)(x+1)≤3×104,解得 x x x 不会超过 300 300 300。
更改一下状态,设 f i , j f_{i,j} fi,j 表示在第 i i i 个点, d d d 变化了 j j j 次所能得到的最大宝石数。
总时间复杂度 O ( 300 n ) O(300n) O(300n)。
Code
#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
inline ll read(){ll x=0,f=1;char c=getchar();while(c<48||c>57){if(c==45)f=0;c=getchar();}while(c>47&&c<58)x=(x<<3)+(x<<1)+(c^48),c=getchar();return f?x:-x;}
const ll maxn=30000,inf=400;
ll n,d,f[maxn+5][inf*2+5],cnt[maxn+5],ans;
signed main(){
n=read(),d=read();
for(ll i=1;i<=n;++i)++cnt[read()];
for(ll i=0;i<=maxn;++i){
for(ll j=-inf;j<=inf;++j){
f[i][inf+j]=LLONG_MIN;
}
}
f[d][inf]=cnt[0]+cnt[d];
for(ll i=d;i<=maxn;++i){
for(ll j=-inf;j<=inf;++j){
if(f[i][inf+j]==LLONG_MIN)continue;
for(ll k=-1;k<=1;++k){
ll len=d+j+k;
if(j+k<-inf||j+k>inf||len<1||i+len>maxn)continue;
f[i+len][inf+j+k]=max(f[i+len][inf+j+k],f[i][inf+j]+cnt[i+len]);
}
}
}
for(ll i=0;i<=maxn;++i){
for(ll j=-inf;j<=inf;++j){
ans=max(ans,f[i][inf+j]);
}
}
printf("%lld",ans);
return 0;
}