POJ2228 Naptime（环形结构上的dp）

题意：一天由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;
}