HDU-4374-One hundred layer

DP+单调队列维护

代码:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int inf=1<<29;
const int maxn=110;
const int maxm=10010;
struct Node
{
    int val;
    int pos;
    Node(){}
    Node(int sval,int spos):val(sval),pos(spos){}
}q[maxm];
int n,m,x,t,sum[maxn][maxm],dp[maxn][maxm];
int main()
{
    while(scanf("%d%d%d%d",&n,&m,&x,&t)!=EOF)
    {
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
            {
                int val;
                scanf("%d",&val);
                sum[i][j]=sum[i][j-1]+val;
            }
        for(int i=1;i<=m;i++)
            dp[1][i]=-inf;
        dp[1][x]=sum[1][x]-sum[1][x-1];
        for(int i=1;i<=t;i++)
        {
            if(i>=x)
                break;
            dp[1][x-i]=sum[1][x]-sum[1][x-i-1];
        }
        for(int i=1;i<=t;i++)
        {
            if(x+i>m)
                break;
            dp[1][x+i]=sum[1][x+i]-sum[1][x-1];
        }
        for(int i=2;i<=n;i++)
        {
            int pre=0,last=0;
            for(int j=1;j<=m;j++)
            {
                while(pre<last&&q[pre].pos<j-t)
                    pre++;
                while(pre<last&&(dp[i-1][j]-sum[i][j-1])>q[last-1].val)
                    last--;
                q[last++]=Node(dp[i-1][j]-sum[i][j-1],j);
                dp[i][j]=sum[i][j]+q[pre].val;
            }
            pre=0,last=0;
            for(int j=m;j>=1;j--)
            {
                while(pre<last&&q[pre].pos-j>t)
                    pre++;
                while(pre<last&&dp[i-1][j]+sum[i][j]>q[last-1].val)
                    last--;
                q[last++]=Node(dp[i-1][j]+sum[i][j],j);
                dp[i][j]=max(dp[i][j],q[pre].val-sum[i][j-1]);
            }
        }
        int ans=-inf;
        for(int i=1;i<=m;i++)
            ans=max(ans,dp[n][i]);
        printf("%d\n",ans);
    }
    return 0;
}


你可能感兴趣的:(dp,单调队列)