原题链接:https://www.luogu.org/problemnew/show/P2627
修剪草坪
题目描述
在一年前赢得了小镇的最佳草坪比赛后,Farm John变得很懒,再也没有修剪过草坪。现在,新一轮的最佳草坪比赛又开始了,Farm John希望能够再次夺冠。
然而,Farm John的草坪非常脏乱,因此,Farm John只能够让他的奶牛来完成这项工作。Farm John有N(1 <= N <= 100,000)只排成一排的奶牛,编号为1…N。每只奶牛的效率是不同的,奶牛i的效率为E_i(0 <= E_i <= 1,000,000,000)。
靠近的奶牛们很熟悉,因此,如果Farm John安排超过K只连续的奶牛,那么,这些奶牛就会罢工去开派对:)。因此,现在Farm John需要你的帮助,计算FJ可以得到的最大效率,并且该方案中没有连续的超过K只奶牛。
输入输出格式
输入格式:
第一行:空格隔开的两个整数 N 和 K
第二到 N+1 行:第 i+1 行有一个整数 E_i
输出格式:
第一行:一个值,表示 Farm John 可以得到的最大的效率值。
输入输出样例
输入样例#1:
5 2
1
2
3
4
5
输出样例#1:
12
题解
我们用
d
p
[
i
]
dp[i]
dp[i]表示前
i
i
i头牛的最优方案,显然有状态转移方程如下;
d
p
[
i
]
=
m
a
x
(
d
p
[
j
−
1
]
+
s
u
m
[
i
]
−
s
u
m
[
j
]
)
(
i
−
k
<
j
<
i
)
dp[i]=max(dp[j-1]+sum[i]-sum[j])(i-k<j<i)
dp[i]=max(dp[j−1]+sum[i]−sum[j])(i−k<j<i)
其中, s u m [ i ] sum[i] sum[i]为前缀和, j j j为不选的奶牛。但是这样的复杂度显然不科学,我们发现这玩意儿只跟 j j j有关且我们每次只取一段区间内的最大值,所以我们可以用单调队列来优化这个 d p dp dp。
代码
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int M=1e5+5;
int n,p,que[M];
ll sum[M],dp[M];
void in()
{
scanf("%d%d",&n,&p);
for(int i=1;i<=n;++i)
scanf("%lld",&sum[i]),sum[i]+=sum[i-1];
}
ll calc(int i){return dp[i-1]-sum[i];}
void ac()
{
int le=0,ri=0,pos;
for(int i=1;i<=n;++i)
{
while(le<ri&&calc(que[ri])<calc(i))ri--;
que[++ri]=i;
while(le<ri&&que[le]<i-p)le++;
pos=que[le];
dp[i]=dp[pos-1]+sum[i]-sum[pos];
}
printf("%lld",dp[n]);
}
int main()
{
in();ac();
return 0;
}