题意:一天由n个小时构成,在第i个小时睡觉能够恢复Ui点体力。有一头牛要休息b个小时,可以不连续,但休息的第1个小时无法恢复体力。前一天的最后一个小时和第二天的第一个小时是连在一起的,求这头牛能恢复的体力最大值。
分析:《算法竞赛进阶指南》P295-297。
代码:
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
using namespace std;
const int N = 4000, M = 4000;
int n, m, f[2][M][2], w[N], ans;
int main (){
scanf ("%d%d", &n, &m);
for (int i = 1; i <= n; i ++)
scanf ("%d", &w[i]);
if (m == 0)
{ printf ("0"); exit (0); }
memset (f, 0x80, sizeof f);
f[1 & 1][0][0] = 0;
f[1 & 1][1][1] = 0;
for (int i = 2; i <= n; i ++)
for (int j = 0; j <= m; j ++)
{
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] + w[i]);
}
ans = max (f[n & 1][m][1], f[n & 1][m][0]);
memset (f, 0x80, sizeof f);
f[1 & 1][1][1] = w[1];
for (int i = 2; i <= n; i ++)
for (int j = 0; j <= m; j ++)
{
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] + w[i]);
}
ans = max (ans, f[n & 1][m][1]);
printf ("%d", ans);
return 0;
}