BZOJ1177&&洛谷P3625 [APIO2009]采油区域

二位前缀和+dp

我们处理五个数组，s,a,b,c,d

s：前缀长度为边长的前缀和

a：当前点的左上的范围内的最大正方形

b：当前点的右上的范围内的最大正方形

c：当前点的左下的范围内的最大正方形

b：当前点的右下的范围内的最大正方形

然后我们可以把图分成

这六种情况

然后取max就好了

代码

//By AcerMo
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int M=2050;
int n,m,e,ans;
int s[M][M],a[M][M],b[M][M],c[M][M],d[M][M];
signed main()
{
    scanf("%d%d%d",&n,&m,&e);int x;
    for (int i=1;i<=n;i++) 
    for (int k=1;k<=m;k++)
    scanf("%d",&x),s[i][k]=x+s[i-1][k]+s[i][k-1]-s[i-1][k-1];
    //读入 
    for (int i=n;i>=e;i--)
    for (int k=m;k>=e;k--)
    s[i][k]-=(s[i-e][k]+s[i][k-e]-s[i-e][k-e]);
    for (int i=e;i<=n;i++)
    for (int k=e;k<=m;k++)
    a[i][k]=max(s[i][k],max(a[i-1][k],a[i][k-1]));
    for (int i=e;i<=n;i++)
    for (int k=m;k>=e;k--)
    b[i][k]=max(s[i][k],max(b[i-1][k],b[i][k+1]));
    for (int i=n;i>=e;i--)
    for (int k=e;k<=m;k++)
    c[i][k]=max(s[i][k],max(c[i+1][k],c[i][k-1]));
    for (int i=n;i>=e;i--)
    for (int k=m;k>=e;k--)
    d[i][k]=max(s[i][k],max(d[i+1][k],d[i][k+1]));
    //预处理 
    for (int i=e;i<=n-e;i++)
    for (int k=e;k<=m-e;k++)
    ans=max(ans,a[i][k]+b[i][k+e]+c[i+e][m]);
    for (int i=e;i<=n-e;i++)
    for (int k=e+e;k<=m;k++)
    ans=max(ans,b[i][k]+d[i+e][k]+a[n][k-e]);
    for (int i=e+e;i<=n;i++)
    for (int k=e;k<=m-e;k++)
    ans=max(ans,c[i][k]+d[i][k+e]+a[i-e][m]);
    for (int i=e;i<=n-e;i++)
    for (int k=e;k<=m-e;k++)
    ans=max(ans,a[i][k]+c[i+e][k]+b[n][k+e]);
    for (int i=e;i<=n;i++)
    for (int k=e+e;k<=m-e;k++)
    ans=max(ans,s[i][k]+a[n][k-e]+b[n][k+e]);
    for (int i=e+e;i<=n-e;i++)
    for (int k=e;k<=m;k++)
    ans=max(ans,s[i][k]+a[i-e][m]+c[i+e][m]);
    //分类讨论 
    cout<<ans;
    return 0;
}