【USACO 2011 Open Gold】修剪草坪(BSOI2940)

【USACO 2011 Open Gold】修剪草坪

Description

在一年前赢得了小镇的最佳草坪比赛后，FJ变得很懒，再也没有修剪过草坪。现在，新一轮的最佳草坪比赛又开始了，FJ希望能够再次夺冠。然而，FJ的草坪非常脏乱，因此，FJ只能够让他的奶牛来完成这项工作。FJ有N(1 <= N <= 100,000)只排成一排的奶牛，编号为1...N。每只奶牛的效率是不同的，奶牛i的效率为E_i(0 <= E_i <= 1,000,000,000)。靠近的奶牛们很熟悉，因此，如果FJ安排超过K只连续的奶牛，那么，这些奶牛就会罢工去开派对:)。因此，现在FJ需要你的帮助，计算FJ可以得到的最大效率，并且该方案中没有连续的超过K只奶牛。

Input

* 第一行：空格隔开的两个整数N和K
* 第二到N+1行：第i+1行有一个整数E_i

Output

* 第一行：一个值，表示FJ可以得到的最大的效率值。

Sample Input

5 2

1

2

3

4

5

输入解释：FJ有5只奶牛，他们的效率为1，2，3，4，5。他们希望选取效率总和最大的奶牛，但是他不能选取超过2只连续的奶牛

Sample Output

12

FJ可以选择出了第三只以外的其他奶牛，总的效率为1+2+4+5=12。

Solution

一道有趣的 USACO（USA~CO：美国co标大赛）。

我们可以将dp变成两个部分方便思考：f[i][0]代表前i头牛不取i的最大值，f[i][1]代表前i头牛取i的最大值。

先将不取第i头牛的情况用上一次的结果赋值，再将距离超过k的队首出队，优化取第i头牛的最大值。

将第i头牛入队，最后ans=max（f[n][0],f[n][1]）

CODE

#include<iostream>
using namespace std;
const int N=100005;
inline long long read() {
	char c;
	int rec=0;
	while((c=getchar())<'0'||c>'9');
	while(c>='0'&&c<='9')rec=rec*10+c-'0',c=getchar();
	return rec;
}
long long s[N];
long long f[N][2],q[N];
int main() {
	int i,n,k,st,en;
	n=read();k=read();
	for(i=1;i<=n;i++){s[i]=read();s[i]+=s[i-1];}
	st=1;en=1;
	for(i=1;i<=n;i++){
		f[i][0]=max(f[i-1][0],f[i-1][1]);
		while(q[st]<i-k&&st<=en)st++;
		f[i][1]=f[q[st]][0]-s[q[st]]+s[i];
		while(f[i][0]-s[i]>f[q[en]][0]-s[q[en]]&&st<=en)en--;
		q[++en]=i;
	}
	cout<<max(f[n][0],f[n][1]);
	return 0;
}