APIO2009 采油区域

Posted 2020-11-30 Dreamunk

挺好玩的一题。

题目

题意：从 (N*M) 的矩阵中选出 (3) 个互不相交的 (K*K) 的正方形，使它们所包含数的和最大。

用上图的六种切法把大矩阵切开，则其中必有一种切法使选出的 (3) 个正方形分别在切出的 (3) 个部分里。

前缀和维护即可。

#include<cstdio>
#define max(a,b)((a)>(b)?(a):(b))
const int N=1502;
int n,m,k,a[N][N],b0[N][N],b1[N][N],b2[N][N],b3[N][N],c0[N],c1[N],ans,tmp;
int main(){
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)
      scanf("%d",&a[i][j]),a[i][j]+=a[i-1][j]-a[i-1][j-1]+a[i][j-1];
    for(int i=n;i>=k;i--)for(int j=m;j>=k;j--)
      a[i][j]-=a[i][j-k]-a[i-k][j-k]+a[i-k][j],
      c0[i]=max(c0[i],a[i][j]),
      c1[j]=max(c1[j],a[i][j]);
    for(int i=k;i<=n;i++)for(int j=k;j<=m;j++)
      b0[i][j]=max(max(b0[i-1][j],b0[i][j-1]),a[i][j]);
    for(int i=k;i<=n;i++)for(int j=m;j>=k;j--)
      b1[i][j]=max(max(b1[i-1][j],b1[i][j+1]),a[i][j]);
    for(int i=n;i>=k;i--)for(int j=k;j<=m;j++)
      b2[i][j]=max(max(b2[i+1][j],b2[i][j-1]),a[i][j]);
    for(int i=n;i>=k;i--)for(int j=m;j>=k;j--)
      b3[i][j]=max(max(b3[i+1][j],b3[i][j+1]),a[i][j]);
    for(int i=k;i+k<=n;i++)for(int j=k;j+k<=m;j++)
      ans=max(ans,b0[i][j]+b1[i][j+k]+b3[i+k][k]),
      ans=max(ans,b1[i][j+k]+b3[i+k][j+k]+b0[n][j]),
      ans=max(ans,b3[i+k][j+k]+b2[i+k][j]+b0[i][m]),
      ans=max(ans,b2[i+k][j]+b0[i][j]+b3[k][j+k]);
    for(int i=k;i+k+k<=n;i++){
      tmp=0;
      for(int j=i+k;j+k<=n;j++)
        tmp=max(tmp,c0[j]),
        ans=max(ans,b0[i][m]+b3[j+k][k]+tmp);
    }
    for(int i=k;i+k+k<=m;i++){
      tmp=0;
      for(int j=i+k;j+k<=m;j++)
        tmp=max(tmp,c1[j]),
        ans=max(ans,b0[n][i]+b3[k][j+k]+tmp);
    }
    printf("%d",ans);
    return 0;
}