状态设计:
- f[小时数(用i&1滚动)][睡觉时长][当前是否在睡觉].
易错点:
- 可以使用滚动数组(i&1).
- 在更新正在睡觉的状态(f[][][1])前需要保证当前总睡觉时间大于1(起码入睡过).
- 在更新f[i&1][j][1]时,可以发现如果上个小时没有睡觉现在就不恢复体力,因此只有在从f[(i-1)&1][j-1][1]尝试进行转移时才需要增加u[i](当前小时的恢复体力量).
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int MAXN=8e3;
int f[2][MAXN][2],u[MAXN];
int main(){
int n,b;
scanf("%d%d",&n,&b);
for(int i=1;i<=n;i++){
scanf("%d",&u[i]);
}
memset(f,0xcf,sizeof(f));
f[1&1][0][0]=f[1&1][1][1]=0;
for(int i=2;i<=n;i++){
for(int j=0;j<=b;j++){
f[i&1][j][0]=max(f[(i-1)&1][j][0],f[(i-1)&1][j][1]);
if(j>=1)f[i&1][j][1]=max(f[(i-1)&1][j-1][0],f[(i-1)&1][j-1][1]+u[i]);
}
}
int ans=max(f[n&1][b][0],f[n&1][b][1]);
memset(f,0xcf,sizeof(f));
f[1&1][1][1]=u[1];
for(int i=2;i<=n;i++){
for(int j=0;j<=b;j++){
f[i&1][j][0]=max(f[(i-1)&1][j][0],f[(i-1)&1][j][1]);
if(j>=1)f[i&1][j][1]=max(f[(i-1)&1][j-1][0],f[(i-1)&1][j-1][1]+u[i]);
}
}
ans=max(ans,f[n&1][b][1]);
printf("%d
",ans);
return 0;
}