题目:修剪草坪
做法:动态规划+单调队列
状态表示:
f
[
i
]
f[i]
f[i] 表示前
i
i
i 头牛合法情况的效率最大值。
状态转移:
1、不选择第
i
i
i 头牛,
f
[
i
]
=
f
[
i
−
1
]
f[i]=f[i-1]
f[i]=f[i−1];
2、选择第 i i i 头牛,那么枚举算上第 i i i 头牛目前一共连续 j j j 头牛的情况,即 f [ i ] = m i n { f [ i − j − 1 ] + s [ i ] − s [ i − j ] , 1 ⩽ j ⩽ k } f[i]=min\{ f[i-j-1]+s[i]-s[i-j], 1\leqslant j \leqslant k\} f[i]=min{f[i−j−1]+s[i]−s[i−j],1⩽j⩽k}.
对上式的解释:实际上目前选的是连续 i − j + 1 i-j+1 i−j+1 到 i i i 这些牛,而且 i − j i-j i−j 一定不能选,因为定义连续 j j j 头牛,所以是用 i − j − 1 i-j-1 i−j−1 的最大值在加上 i − j + 1 i-j+1 i−j+1 到 i i i 这些牛的前缀和。
最后取情况一和情况二的最小值。
所以在维护长度为 k k k 的滑动窗口最值时用单调队列优化。
一开始队列不为空, f [ 0 ] = 0 f[0]=0 f[0]=0.
代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=100010;
typedef long long ll;
ll s[N],f[N];
int n,k;
int q[N];
ll g(int i){
if(!i) return 0;
else return f[i-1]-s[i];
}
int main(){
cin>>n>>k;
for(int i=1;i<=n;i++) scanf("%lld",&s[i]),s[i]+=s[i-1];
int hh=0,tt=0;
for(int i=1;i<=n;i++){
if(q[hh]<i-k) hh++; //这里不是<=是因为q[hh]实际控制的不是队头,而是对头的前一个
//f[i]表示前i头牛合法的最大效率
f[i]=max(f[i-1],g(q[hh])+s[i]);
while(hh<=tt && g(q[tt])<=g(i)) tt--;
q[++tt]=i;
}
cout<<f[n]<<endl;
return 0;
}
这道题和烽火旅行那道题有异曲同工的感觉,不过在状态定义的时候不一样,因此此题最后不需要扫描一遍而是直接输出 f [ n ] f[n] f[n] 即可。如果有时间,会换那种题的表示方法再试一试。