POJ2228 Naptime(원형 구조의 dp)

1448 단어 dp

제목: 하루에 n시간으로 구성되어 있으며, i시간에 잠을 자면 Ui점 체력을 회복할 수 있다.소 한 마리가 b시간을 쉬어야 하기 때문에 연속하지 않아도 되지만 쉬는 1시간은 체력을 회복할 수 없다.전날 마지막 시간과 이튿날 첫 시간은 이어져 이 소가 회복할 수 있는 체력 최대치를 구한다.


분석: P295-297.


코드:

#include 
#include 
#include 
#include 

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;
}

좋은 웹페이지 즐겨찾기