[BZOJ 1177][Apio2009]Oil：DP

点击这里查看原题

s[i][j]表示以(i,j)为右下角的正方形的面积
因为用两条直线可以把矩形分为三个部分且使三个正方形分别在三个矩形内，所以枚举每种分割情况，用分割后每块内的面积最大的正方形的值更新答案
具体分割情况如图：
1.可以通过两条互相垂直的线分成三块，如下图

2.可以通过两条平行的平行的线分成三块，如下图

/*
User:Small
Language:C++
Problem No.:1177
*/
#include<bits/stdc++.h>
#define ll long long
#define inf 999999999
using namespace std;
const int M=2505;
int n,m,s[M][M],a[M][M],b[M][M],c[M][M],d[M][M],ans,k;
int main(){
    freopen("data.in","r",stdin);//
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            int x;
            scanf("%d",&x);
            s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+x;
        }
    }
    for(int i=n;i>=k;i--)
        for(int j=m;j>=k;j--)
            s[i][j]+=s[i-k][j-k]-s[i-k][j]-s[i][j-k];

    for(int i=k;i<=n;i++)
        for(int j=k;j<=m;j++) a[i][j]=max(s[i][j],max(a[i-1][j],a[i][j-1]));
    for(int i=k;i<=n;i++)
        for(int j=m;j>=k;j--) b[i][j]=max(s[i][j],max(b[i-1][j],b[i][j+1]));
    for(int i=n;i>=k;i--)
        for(int j=k;j<=m;j++) c[i][j]=max(s[i][j],max(c[i+1][j],c[i][j-1]));
    for(int i=n;i>=k;i--)
        for(int j=m;j>=k;j--) d[i][j]=max(s[i][j],max(d[i+1][j],d[i][j+1]));

    for(int i=k;i<=n-k;i++)
        for(int j=k;j<=m-k;j++) ans=max(ans,a[i][j]+b[i][j+k]+c[i+k][m]);
    for(int i=k;i<=n-k;i++)
        for(int j=2*k;j<=m;j++) ans=max(ans,b[i][j]+d[i+k][j]+a[n][j-k]);
    for(int i=2*k;i<=n;i++)
        for(int j=k;j<=m-k;j++) ans=max(ans,c[i][j]+d[i][j+k]+a[i-k][m]);
    for(int i=k;i<=n-k;i++)
        for(int j=k;j<=m-k;j++) ans=max(ans,a[i][j]+c[i+k][j]+b[n][j+k]);

    for(int i=2*k;i<=n-k;i++)
        for(int j=k;j<=m;j++) ans=max(ans,a[i-k][m]+s[i][j]+c[i+k][m]);
    for(int i=k;i<=n;i++)
        for(int j=2*k;j<=m-k;j++) ans=max(ans,a[n][j-k]+s[i][j]+b[n][j+k]);

    printf("%d\n",ans);
    return 0;
}