1.确定子状态:选完了前i个,当前剩下j个
2.dp柿子:dp[i][j]表示如上含义
3.递推公式:dp[i][j]=maxdp[i-1][j-1],j..j+s-1;
4.去重(无)
这里对区间最值做一个单调队列的优化(从后往前的滑动区间)
先理清写出思路!
好好写代码
#include<iostream>
#include<cstring>
#include<vector>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll inf=100000000000000000;
const int mod=998244353;
const int nn=1e5+50;
const double eps=1e-6;
inline int read() {
int x=0,f=1;
char c=getchar();
while(c<'0'||c>'9'){if(c=='-') f=-1;c=getchar();}
while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
return x*f;
}
inline void write(int x) {
if(x<0) putchar('-'),x=-x;
if(x>9) write(x/10);
putchar(x%10+'0');
}
struct one{
int x,y;
};
bool cmp(one a,one b)
{
if(a.x==b.x)
return a.y>b.y;
return a.x>b.x;
}
ll dp[5005][5005];
ll a[5005];
vector <int> v;
int main()
{
ll n,w,s;
n=read();
w=read(),s=read();
for(int i=1;i<=n;++i)
{
a[i]=read();
}
for(int i=0;i<=n;++i)
{
for(int j=0;j<=w;++j)
{
dp[i][j]=-inf;
}
}
dp[0][0]=0;
int b[5005];
int l,r;
ll anss[5005];
for(ll i=1;i<=n;++i)
{
for(ll j=0;j<=w;++j)
{
if(j==0||j>i)
{
dp[i][j]=-inf;
continue;
}
dp[i][j]=anss[j-1]+j*a[i];
}
l=w+1,r=w;
dp[i][0]=-inf;
for(ll j=w;j>=0;--j)
{
while(l<=r&&dp[i][b[l]]<dp[i][j]) l++;
b[--l]=j;
if(b[r]-j>s) r--;
anss[j]=dp[i][b[r]];
// if(i==5&&j==4)
// cout<<anss[j]<<endl;
}
}
ll ans=-inf;
// for(int i=0;i<=n;++i)
// {
// for(int j=0;j<=w;++j)
// cout<<dp[i][j]<<" ";
// cout<<endl;
// }
for(int i=1;i<=w;++i)
{
if(dp[n][i]>ans)
ans=dp[n][i];
}
cout<<ans;
return 0;
}