- 休息时间
在某个星球上,一天由 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。
不难发现,这道题目是一个环形的问题。对于一个环形的问题,第一想法就是化环为链,但是这一题提供了一种新的思路。
我们分情况讨论,先把题目当做没有环的情况。再考虑有环的情况会多出哪一种可能性。这一种可能性就是1号时间休息。1号时间休息的前提条件就是n号时间也必须休息。所以我们只用把链的答案和强行取第一个和最后一个时间的答案取一个max.
考虑空间上的优化,每一个阶段的dp值只能从上一个阶段得到,那么我们可以采取滚动数组优化空间。但是要注意一个细节,一般的写法是f[i]=max(f[i].f[i-j]+w[j])。但是这里的f[i]表示的是上上个阶段,所以要从上一个阶段的状态继承。
#include<bits/stdc++.h>
using namespace std;
int n,m;
int a[4000];
int f[3][4000][3];
int ff[3][4000][3];
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
memset(f,-0x3f,sizeof(f));
memset(ff,-0x3f,sizeof(ff));
f[1][0][0]=0;f[1][1][1]=0;
ff[1][1][1]=a[1];
for(int i=2;i<=n;i++){
for(int j=0;j<=min(i,m);j++){
f[i%2][j][0]=max(f[(i+1)%2][j][0],f[(i+1)%2][j][1]);
if(j>=1) f[i%2][j][1]=max(f[(i+1)%2][j-1][0],f[(i+1)%2][j-1][1]+a[i]);
ff[i%2][j][0]=max(ff[(i+1)%2][j][0],ff[(i+1)%2][j][1]);
if(j>=1) ff[i%2][j][1]=max(ff[(i+1)%2][j-1][0],ff[(i+1)%2][j-1][1]+a[i]);
}
}
printf("%d\n",max(ff[n%2][m][1],max(f[n%2][m][0],f[n%2][m][1])));
}