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