题目描述
传送门
题解
f[i][j][0]
表示到第i个一共选了j段,第i个不选的最大价值
f[i][j][1]
表示到第i个一共选了j段,第i个必选的最大价值
转移方程
f[i][j][0]=max(f[i−1][j][0],f[i−1][j][1])
f[i][j][1]=max(f[i−1][j−1][0],f[i−1][j−1][1]+val[i])
考虑环的问题,我们可以强制第一个选并且不作为入睡阶段,然后枚举后面与之相邻的个数,计算答案。
因为卡内存,所以需要滚动数组。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 4003
using namespace std;
int f[3][N][3],n,m,val[N],sum[N];
int main()
{
freopen("a.in","r",stdin);
freopen("my.out","w",stdout);
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) scanf("%d",&val[i]);
for (int i=n;i>=1;i--) sum[i]=sum[i+1]+val[i];
memset(f,128,sizeof(f)); int INF=f[0][0][0];
f[0][0][0]=0;
for (int i=1;i<=n;i++) {
int t=i%2;
for (int j=0;j<=min(m,i);j++) f[t][j][0]=f[t][j][1]=INF;
f[t][0][0]=0;
for (int j=1;j<=min(m,i);j++) {
f[t][j][0]=max(f[t][j][0],f[t^1][j][0]);
f[t][j][0]=max(f[t][j][0],f[t^1][j][1]);
f[t][j][1]=max(f[t][j][1],f[t^1][j-1][0]);
f[t][j][1]=max(f[t][j][1],f[t^1][j-1][1]+val[i]);
}
}
int ans=max(f[n%2][m][0],f[n%2][m][1]);
memset(f,128,sizeof(f));
f[1][1][1]=val[1];
for (int i=2;i<=n;i++){
int t=i%2;
for (int j=0;j<=min(m,i);j++) f[t][j][0]=f[t][j][1]=INF;
for (int j=1;j<=min(m,i);j++) {
f[t][j][0]=max(f[t][j][0],f[t^1][j][0]);
f[t][j][0]=max(f[t][j][0],f[t^1][j][1]);
f[t][j][1]=max(f[t][j][1],f[t^1][j-1][0]);
f[t][j][1]=max(f[t][j][1],f[t^1][j-1][1]+val[i]);
}
int t1=n-i+1;
if (m>=t1) ans=max(ans,f[t][m-t1][0]+sum[i+1]);
}
printf("%d\n",ans);
}