题目地址

状态设计：

  • 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\n",ans);
	return 0;
}