[USACO 2010 Ope G]Cow Hopscotch（单调队列优化dp）

[USACO 2010 Ope G]Cow Hopscotch（单调队列优化dp）

• 题目链接
• 题意：给定一个 n 个格子，每个格子有一个价值，一头牛一开始在 0 格子，每次最多跳 k 个，每当它跳到某个格子上，它就能够获得该格子的价值，并且它随时可以往后跳，但是有两个限制，一个就是不能再跳已经跳过的格子，再就是返回时若要跳到 i-1,则 i 在向前跳时必须跳过，直到它跳到 0 格子结束，求它最多能获得多少价值。
• 解题思路：首先这是一个dp问题，但是首先要解决往回跳的影响，又有题目第二个限制，所以将dp[i]定义为走到i直接返回i-1的话最能赚多少钱，在计算dp[i]时也计算上i-1的值，这样就把返回的值都给算上了。dp[i] =a[i]+a[i-1]+ max{dp[j] | i-j >= k && j <= i-2}， 这个状态方程很容易写出来，但是它是不对的，为什么是不对的呢，因为如果 i-k ~ i-2 这些格子中可能它在向前的时候也跳过了，只不过在往后跳的时候没跳过，这样的格子我们没有计算过，再考虑这些格子肯定都是正数，我们不可能去跳负数。状态方程就改变成了 dp[i] = a[i]+a[i-1]+max{dp[j] + sum[i-2]-sum[j] | i-j >= k && j <= i-2}，这里 sum[i] 是表示 1~i 这个区间内的所有正数的和。
• 这样的复杂度时O(nk)，且K范围较大，所以还要用单调队列维护[i-k,i-2]范围内的最大dp[j]-sum[j]。同时还要考虑特殊情况k>=n，能够把所有正数都拿走，先初始化ans=sum[min(n,k)]。真实情况中🐂并不是走到dp[i]就直接返回，它可以再往前跳时把i+1到i-1+k的整数都跳到，然后直接跳回i-1，所以答案是ans=max(ans,{dp[i]+sum[min(i-1+k,n)]-sum[i] | 1<=i<=n} }
• 代码：

#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>
#include<deque>
#include <queue>
#include <stack>
#include <set>
#include <bitset>
#include <cstdlib>
#include <map>
#include <cstdio>
#include <cstring>
#include <iomanip>
#define INF 0x3f3f3f3f
//#define INF ~0u//2^32-1
#define mod 1000000007
//#define mod 998244353
#define pi acos(-1)
#define ll long long
using namespace std;
const int N=3e5+5;
const double eps=1e-10;

ll a[N],dp[N],sum[N];   //dp[i]表示走到i直接返回i-1的话能赚多少钱
int main(void)
{
    int n,k;
    scanf("%d %d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        sum[i]=sum[i-1]+(a[i]>0?a[i]:0);    //正整数前缀和
    }
    deque<int> q;
    dp[1]=a[1];
    q.push_back(1);
    for(int i=2,j=0;i<=n;i++,j++)
    {
        //用单调队列优化寻找[i-k，i-2]区间内最大dp[j]-sum[j]的过程;
        if(!q.empty()&&i-q.front()>k)
            q.pop_front();
        ll val=dp[j]-sum[j];
        while(!q.empty()&&val>=dp[q.back()]-sum[q.back()])
            q.pop_back();
        q.push_back(j);

        //dp方程：dp[i] = max{dp[j] + sum[i-2]-sum[j] | i-j >= k && j <= i-2}
        dp[i]=dp[q.front()]+a[i]+a[i-1]+sum[i-2]-sum[q.front()];
    }
    ll ans=sum[min(n,k)];
    for(int i=1;i<=n;i++)
    {
        ans=max(ans,dp[i]+sum[min(n,i-1+k)]-sum[i]);    //实际上从i到i-1+k位都可在前进时走到，只是返回时直接走到i-1,所以加上该区间的正整数和
    }
    printf("%lld\n",ans);
    return 0;
}

• 参考博客