【动态规划】睡觉

PROBLEM 3 睡觉

 

问题描述：

为了提高程序解题能力，勤奋努力的QQ天天锯题到深夜，导致睡眠严重不足，可NOIP决赛就要来临了，必须要有良好的状态才行啊，因此QQ决定准备拿出一天时间，好好补补觉。

他把这一天等分成了n个时间段，在每个时间段睡觉能获得精神点数不尽相同，在第i段时间能获得V[i]的精神点数。

由于勤奋的QQ觉得整天都睡太堕落了，他决定最多只能睡m个时间段。至于其他的时间吗...那自然是勤奋地锯题...

有两点事情要特别提出注意：

1.QQ不可能一上床马上睡着，他在连续一段睡觉时间的第一个时间段不能获得此时的V。也就是说如果他在i...j中的所有时间都休息了，获得的精神点数为V[i+1]+...+V[j]。

2.所有的时间段呈环形分布，也就是说第n个时间段之后为第1个时间段。

要求的自然是QQ最多能获得的精神点数之和。

 

输入格式：

第一行两个正整数n,m，意义如题所述

接下来n行，每行一个非负整数V[i]

 

输出格式：

一行，表示QQ最多能获得的精神点数之和

 

输入样例：

5 3

2

0

3

1

4

 

输出样例：

  6

 

样例解释：

选择4、5、1三个时间段休息，在5、1时睡着，最大值为4+2=6

 

数据范围：

对于20%的数据n<=20

对于50%的数据n<=200

对于100%的数据n<=5000,m<=n,V[i]<=10000

这道题和最近做的一道环形动规比较相似，断链方式都是分别讨论起点的状态。可以降低一维。

如果起点要睡，则从后面开始睡睡到起点不停止。

如果起点不睡，则按照一般的方法转移。

但是注意滚动，不滚动的话，因为赋初值会超时两组。

#include <cstdio>
#include <cstring>
#include <string>
#define max(a,b) ((a)>(b)?(a):(b))

long v[5010];
long f[5010][2];

long getint()
{
	long rs=0;bool sgn=1;char tmp;
	do tmp=getchar();
	while (!isdigit(tmp)&&tmp-'-');
	if (tmp=='-'){tmp=getchar();sgn=0;}
	do rs=(rs<<3)+(rs<<1)+tmp-'0';
	while (isdigit(tmp=getchar()));
	return sgn?rs:-rs;
}

int main()
{
	freopen("sleep.in","r",stdin);
	freopen("sleep.out","w",stdout);

	long n = getint();
	long m = getint();

	if (m == 0)
	{
		printf("0");
		return 0;
	}

	for (long i=1;i<n+1;i++)
		v[i+n] = v[i] = getint();

	if (n == m)
	{
		long sum = 0;
		long miv = 0x3f3f3f3f;
		for (long i=1;i<n+1;i++)
		{
			if (v[i] < miv)
				miv = v[i];
			sum += v[i];
		}
		printf("%ld",sum-miv);
		return 0;
	}

	long ans = -0x3f3f3f3f;

	//--------------------------------- 一时刻要睡到末尾完后接着睡1
	memset(f,~0x3f,sizeof f);
	f[1][1&1] = 0;
	f[2][2&1] = v[2];
	for (long i=3;i<n+1;i++)
		f[i][2&1] = 0;
	for (long j=3;j<m+1;j++)
	{
		long great = -0x3f3f3f3f;
		for (long i=1;i<n+1;i++)
		{
			great = max(great,f[i-1][(j&1)^1]);
			f[i][j&1] = max(great,f[i-1][(j&1)^1]+v[i]);
		}
	}
	for (long i=1;i<n;i++)
		ans = max(ans,f[i][m&1]);
	ans = max(ans,f[n][m&1]+v[1]);
	//--------------------------------- 一时刻不睡
	memset(f,~0x3f,sizeof f);
	f[1][1&1] = -0x3f3f3f3f;
	for (long i=2;i<n+1;i++)
		f[i][1&1] = 0;
	for (long j=2;j<m+1;j++)
	{
		long great = -0x3f3f3f3f;
		for (long i=1;i<n+1;i++)
		{
			great = max(great,f[i-1][(j&1)^1]);
			f[i][j&1] = max(great,f[i-1][(j&1)^1]+v[i]);
		}
	}
	for (long i=1;i<n+1;i++)
		ans = max(ans,f[i][m&1]);
	//----------------------------------
	printf("%ld",ans);
	return 0;
}