BZOJ1177: Oil 题解

Description

采油区域 Siruseri政府决定将石油资源丰富的Navalur省的土地拍卖给私人承包商以建立油井。被拍卖的整块土地为一个矩形区域，被划分为M×N个小块。 Siruseri地质调查局有关于Navalur土地石油储量的估测数据。这些数据表示为M×N个非负整数，即对每一小块土地石油储量的估计值。 为了避免出现垄断，政府规定每一个承包商只能承包一个由K×K块相连的土地构成的正方形区域。 AoE石油联合公司由三个承包商组成，他们想选择三块互不相交的K×K的区域使得总的收益最大。 例如，假设石油储量的估计值如下： 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 8 8 8 8 8 1 1 1 1 8 8 8 8 8 1 1 1 1 8 8 8 8 8 1 1 1 1 1 1 1 8 8 8 1 1 1 1 1 1 1 1 8 8 8 1 1 1 1 1 1 9 9 9 1 1 1 1 1 1 9 9 9 如果K = 2, AoE公司可以承包的区域的石油储量总和为100, 如果K = 3, AoE公司可以承包的区域的石油储量总和为208。 AoE公司雇佣你来写一个程序，帮助计算出他们可以承包的区域的石油储量之和的最大值。

Input

输入第一行包含三个整数M, N, K，其中M和N是矩形区域的行数和列数，K是每一个承包商承包的正方形的大小（边长的块数）。接下来M行，每行有N个非负整数表示这一行每一小块土地的石油储量的估计值

Output

输出只包含一个整数，表示AoE公司可以承包的区域的石油储量之和的最大值。

Sample Input

9 9 3
1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1
1 8 8 8 8 8 1 1 1
1 8 8 8 8 8 1 1 1
1 8 8 8 8 8 1 1 1
1 1 1 1 8 8 8 1 1
1 1 1 1 1 1 8 8 8
1 1 1 1 1 1 9 9 9
1 1 1 1 1 1 9 9 9

Sample Output

208

---

先吐槽一波这题数据有毒，好像行数和列数不太对这类的，反正不要用快读，用scanf就能过
难度其实不大，仔细想想可以发现，一定可以画一条直线，使得直线一边有一个矩形，另一边有两个矩形
这样我们只要预处理在某条直线的上/下/左/右的区域选两个矩形，和的最大值，然后枚举最后一个矩形，就可以计算了
对于预处理，考虑计算有一个矩形正好贴在直线上的选两个矩形的最大值，这样做完以后做一个前缀max就好了
然后我们发现做这个预处理，我们还要预处理在某个点左上/左下/右上/右下的区域里选一个矩形的最大值，这个可以通过前缀max预处理
然后一波算就好了，讲不太清楚，可以看代码

#include <bits/stdc++.h>
using namespace std;

#define LL long long
#define LB long double
#define ull unsigned long long
#define x first
#define y second
#define pb push_back
#define pf push_front
#define mp make_pair
#define Pair pair<int,int>
#define pLL pair<LL,LL>
#define pii pair<double,double>

const int INF=1.5e9;
const LL LINF=2e16;
const int magic=348;
const int MOD=998244353;
const double eps=1e-10;
const double pi=acos(-1);

inline int getint()
{
    bool f;char ch;int res;
    while (!isdigit(ch=getchar()) && ch!='-') {}
    if (ch=='-') f=false,res=0; else f=true,res=ch-'0';
    while (isdigit(ch=getchar())) res=res*10+ch-'0';
    return f?res:-res;
}

const int MAXN=1500;

int n,m,k;
int a[MAXN+48][MAXN+48],sum[MAXN+48][MAXN+48],b[MAXN+48][MAXN+48];
int maxn1[4][MAXN+48][MAXN+48],maxn2[4][MAXN+48];

inline int query(int x1,int y1,int x2,int y2)
{
    return sum[x2][y2]-sum[x1-1][y2]-sum[x2][y1-1]+sum[x1-1][y1-1];
}

int main ()
{
    int i,j,p;
    scanf("%d%d%d",&n,&m,&k);
    for (i=1;i<=n;i++)
        for (j=1;j<=m;j++)
            scanf("%d",&a[i][j]);
    for (i=1;i<=n;i++)
    {
        sum[i][0]=0;
        for (j=1;j<=m;j++) sum[i][j]=sum[i][j-1]+a[i][j];
    }
    for (i=2;i<=n;i++)
        for (j=1;j<=m;j++)
            sum[i][j]+=sum[i-1][j];
    for (i=1;i<=n-k+1;i++)
        for (j=1;j<=m-k+1;j++)
            b[i][j]=query(i,j,i+k-1,j+k-1);
    for (p=0;p<=3;p++)
        for (i=0;i<=n+1;i++)
            for (j=0;j<=m+1;j++)
                maxn1[p][i][j]=-INF;
    for (i=1;i<=n-k+1;i++)
        for (j=1;j<=m-k+1;j++)
        {
            maxn1[0][i+k-1][j+k-1]=max(maxn1[0][i][j],b[i][j]);
            maxn1[1][i+k-1][j]=max(maxn1[1][i+k-1][j],b[i][j]);
            maxn1[2][i][j+k-1]=max(maxn1[2][i][j+k-1],b[i][j]);
            maxn1[3][i][j]=max(maxn1[3][i][j],b[i][j]);
        }
    for (i=1;i<=n;i++)
        for (j=2;j<=m;j++)
        {
            maxn1[0][i][j]=max(maxn1[0][i][j],maxn1[0][i][j-1]);
            maxn1[2][i][j]=max(maxn1[2][i][j],maxn1[2][i][j-1]);
        }
    for (i=1;i<=n;i++)
        for (j=m-1;j>=1;j--)
        {
            maxn1[1][i][j]=max(maxn1[1][i][j],maxn1[1][i][j+1]);
            maxn1[3][i][j]=max(maxn1[3][i][j],maxn1[3][i][j+1]);
        }
    for (i=2;i<=n;i++)
        for (j=1;j<=m;j++)
        {
            maxn1[0][i][j]=max(maxn1[0][i][j],maxn1[0][i-1][j]);
            maxn1[1][i][j]=max(maxn1[1][i][j],maxn1[1][i-1][j]);
        }
    for (i=n-1;i>=1;i--)
        for (j=1;j<=m;j++)
        {
            maxn1[2][i][j]=max(maxn1[2][i][j],maxn1[2][i+1][j]);
            maxn1[3][i][j]=max(maxn1[3][i][j],maxn1[3][i+1][j]);
        }
    for (i=0;i<=3;i++)
        for (j=1;j<=max(n,m);j++)
            maxn2[i][j]=-INF;
    for (i=k;i<=n;i++)
        for (j=1;j<=m-k+1;j++)
            maxn2[0][i]=max(maxn2[0][i],b[i-k+1][j]+max(maxn1[0][i-k][m],max(maxn1[0][i][j-1],maxn1[1][i][j+k])));
    for (i=2;i<=n;i++) maxn2[0][i]=max(maxn2[0][i],maxn2[0][i-1]);
    for (i=n-k+1;i>=1;i--)
        for (j=1;j<=m-k+1;j++)
            maxn2[1][i]=max(maxn2[1][i],b[i][j]+max(maxn1[2][i+k][m],max(maxn1[2][i][j-1],maxn1[3][i][j+k])));
    for (i=n-1;i>=1;i--) maxn2[1][i]=max(maxn2[1][i],maxn2[1][i+1]);
    for (j=k;j<=m;j++)
        for (i=1;i<=n-k+1;i++)
            maxn2[2][j]=max(maxn2[2][j],b[i][j-k+1]+max(maxn1[0][n][j-k],max(maxn1[0][i-1][j],maxn1[2][i+k][j])));
    for (j=2;j<=m;j++) maxn2[3][j]=max(maxn2[3][j],maxn2[3][j-1]);
    for (j=m-k+1;j>=1;j--)
        for (i=1;i<=n-k+1;i++)
            maxn2[3][j]=max(maxn2[3][j],b[i][j]+max(maxn1[1][n][j+k],max(maxn1[1][i-1][j],maxn1[3][i+k][j])));
    for (j=m-1;j>=1;j--) maxn2[3][j]=max(maxn2[3][j],maxn2[3][j+1]);
    int ans=0;
    for (i=1;i<=n-k+1;i++)
        for (j=1;j<=m-k+1;j++)
            ans=max(ans,max(max(b[i][j]+maxn2[0][i-1],b[i][j]+maxn2[1][i+k]),max(b[i][j]+maxn2[2][j-1],b[i][j]+maxn2[3][j+k])));
    printf("%d\n",ans);
    return 0;
}