288. 休息时间

题目

在某个星球上，一天由 N 个小时构成，我们称0点到1点为第1个小时、1点到2点为第2个小时，以此类推。
在第 i 个小时睡觉能够恢复Ui点体力。
在这个星球上住着一头牛，它每天要休息B个小时。
它休息的这B个小时不一定连续，可以分成若干段，但是在每段的第一个小时，它需要从清醒逐渐入睡，不能恢复体力，从下一个小时开始才能睡着。
为了身体健康，这头牛希望遵循生物钟，每天采用相同的睡觉计划。
另外，因为时间是连续的，即每一天的第N个小时和下一天的第1个小时是相连的（N点等于0点），这头牛只需要在每N个小时内休息够B个小时就可以了。
请你帮忙给这头牛安排一个睡觉计划，使它每天恢复的体力最多。

输入格式
第1行输入两个空格隔开的整数N和B。
第2…N+1行，第 i+1 行包含一个整数Ui。

输出格式
输出一个整数，表示恢复的体力值。

数据范围
3≤N≤3830
2≤B<N
0≤Ui≤200000
输入样例：
5 3
2
0
3
1
4
输出样例：
6
样例解释
这头牛每天3点入睡，睡到次日1点，即[1,4,2]时间段休息，每天恢复体力值最大，为0+4+2=6。

这是一道只可以从一个地方断开环的环形dp。
设 f[i][j][0]表示在前i小时休息j小时且第i小时不休息。
f[i][j][1]表示在前i小时休息j小时且第i小时休息。
1将环断开 即不考虑前一天
强行让第一个小时不进入深度休息。
那么方程就十分显然了

f[1][1][1]=0;
f[1][0][0]=0;

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]+a[i]);
ans=max(f[n][b][0],f[n][b][1]);

2将环接上 即考虑前一天最后一小时的休息情况
强行让第一个小时进入深度睡眠

f[1][1][1]=a[1];

最后的答案是f[n][b][1]

最后的答案是两种情况的较大值

#include<bits/stdc++.h>
using namespace std;
const int ll=3835;
int n,b,t;
int a[ll],f[2][ll][2],ans;

int main(){
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&n,&b);
		for(int i=1;i<=n;i++)	
			scanf("%d",&a[i]);
		memset(f,-0x3f,sizeof(f));
		f[1][1][1]=0;
		f[1][0][0]=0;
		for(int i=2;i<=n;i++)
			for(int j=0;j<=b;j++){
				if(i<j) continue;
				f[i&1][j][0]=max(f[(i-1)&1][j][0],f[(i-1)&1][j][1]);
				if(j-1>=0)
				f[i&1][j][1]=max(f[(i-1)&1][j-1][0],f[(i-1)&1][j-1][1]+a[i]);
			}
		ans=max(f[n&1][b][0],f[n&1][b][1]);
		
		memset(f,-0x3f,sizeof(f));
		f[1][1][1]=a[1];
		for(int i=2;i<=n;i++)
			for(int j=0;j<=b;j++){
				if(i<j) continue;
				f[i&1][j][0]=max(f[(i-1)&1][j][0],f[(i-1)&1][j][1]);
				if(j-1>=0)
				f[i&1][j][1]=max(f[(i-1)&1][j-1][0],f[(i-1)&1][j-1][1]+a[i]);
			}
		ans=max(ans,f[n&1][b][1]);
		printf("%d\n",ans);
    }
	return 0;
}