【HDU4374】One hundred layer-单调队列优化DP

测试地址：One hundred layer
题目大意：一座塔有N层，每一层有一排M个位置，要从第一层的第X个位置出发，每一层只能向某一个方向走动至多T步，然后就要跳到下一层的与该层跳下位置对应的那个位置。每个位置上都有一个分数，最终得分就是所有走过的位置上分数的和，求最大的最终得分。
做法：本题是一个单调队列优化DP的题目。
这一题是DP应该能很显然地看出来，设f(i,j)为从第i层第j个位置开始走，一直到达最底层的最大得分，那么有状态转移方程：
f(i,j)=max(max{f(i+1,k)+w(i,k,j)|j−T≤k≤j},max{f(i+1,k)+w(i,j,k)|j≤k≤j+T})
其中w(i,j,k)为第i行第j到第k个位置的分数之和，即∑kp=jaip。这个方程应该很容易理解，里面的两个max分别表示向左走和向右走的最大得分，两者再取个max就是总体上的最大得分了。
w可以用前缀和O(NM)预处理出来，那么上述方程就是O(NM2)的。显然会炸，需要考虑优化。
考虑到向左走和向右走两种情况是对称的，所以我们这里只讨论向左走的情况。我们处理出了前缀和sum(i,j)=∑jk=1aik，那么上面的式子可以写成：
max{f(i+1,k)+sum(i,j)−sum(i,k−1)|j−T≤k≤j}
我们发现sum(i,j)在求f(i,j)时可以算作常量，可以提出括号，所以我们就是要求：
max{f(i+1,k)−sum(i,k−1)|j−T≤k≤j}
观察发现，f(i+1,k)−sum(i,k−1)是由变量k所唯一确定的常量，而k的取值范围大小恒定，显然可以使用单调队列优化，这样方程的复杂度就是O(NM)的了。
对于向右走的情况同理，这里就不赘述了，最后算法的时间复杂度为O(NM)，完美解决了这一题。
最后还需要注意的是，输入可能有多组测试数据，小心不要被坑了……
以下是本人代码：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define inf 1000000000
using namespace std;
int n,m,x,t;
int a[110][10010],f[110][10010],lsum[110][10010],rsum[110][10010];
int q[10010],head,tail;

int main()
{
    while(scanf("%d%d%d%d",&n,&m,&x,&t)!=EOF)
    {
        memset(lsum,0,sizeof(lsum));
        memset(rsum,0,sizeof(rsum));
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                scanf("%d",&a[i][j]);
        for(int i=1;i<=m;i++) f[n+1][i]=0;
        for(int i=1;i<=n;i++)
        {
            lsum[i][0]=rsum[i][m+1]=0;
            for(int j=1;j<=m;j++) lsum[i][j]=lsum[i][j-1]+a[i][j];
            for(int j=m;j>=1;j--) rsum[i][j]=rsum[i][j+1]+a[i][j];
        }
        for(int i=n;i>=1;i--)
        {
            head=1,tail=0;
            for(int j=1;j<=m;j++)
            {
                f[i][j]=-inf;
                while(head<=tail&&q[head]<j-t) head++;
                while(head<=tail&&f[i+1][j]-lsum[i][j-1]>=f[i+1][q[tail]]-lsum[i][q[tail]-1]) tail--;
                q[++tail]=j;
                f[i][j]=max(f[i][j],f[i+1][q[head]]+lsum[i][j]-lsum[i][q[head]-1]);
            }
            head=1,tail=0;
            for(int j=m;j>=1;j--)
            {
                while(head<=tail&&q[head]>j+t) head++;
                while(head<=tail&&f[i+1][j]-rsum[i][j+1]>=f[i+1][q[tail]]-rsum[i][q[tail]+1]) tail--;
                q[++tail]=j;
                f[i][j]=max(f[i][j],f[i+1][q[head]]+rsum[i][j]-rsum[i][q[head]+1]);
            }
        }
        printf("%d\n",f[1][x]);
    }

    return 0;
}