Vijos P1742 扑克の阵

Vijos P1742 扑克の阵

---

题目

背景

一片幻影闪过，眼花缭乱，什么都看不清……
终于，一切平静了下来。一袭白衣，礼帽，斗篷，手套，皮鞋，闪着光的镜片下出现一张英俊的脸庞（啊！！OIBH组织众女生花痴ing）……
怪盗基德浅浅一笑，想要困住我，还差点本事呢！（啊！！能杀人的笑……OIBH组织众女生继续花痴ing）那么，女士们，先生们，就让我，举世闻名的魔术师，为各位献上一场独一无二的魔术秀吧！
掌声雷动……掌声雷动……撒花……撒花……
第1009号小兵屁颠屁颠从门口跑过来，刚想开口喊：“抓基德啊！！”却不禁被眼前华丽的魔术吸引了……
描述

在越过了第一道关卡之后，KID暂时摆脱了OIBH组织的追兵。基德为了不被打扰，决定送给OIBH组织一份大礼：那就是扑克阵法。基德的扑克阵法是扑克牌组成的N*N的大方阵，要解开这个阵法的要诀就是，拿走两个互不重叠的K*K的方阵中的扑克牌，使得这2*k^2张扑克牌的总分最大，从而使整个阵法失去主要能量而土崩瓦解。但是，基德自己在偷完回去的路上也会碰到这个阵法——为了不至于自己把自己困住——基德需要你帮助他编程解决这个阵法。
格式

输入格式

第1行两个数n，k。
第2行到第n+1行，每行n个非负整数，表示整个扑克阵法每张牌的分值。
输出格式

一个数，表示最大能拿走的总分。
样例1

样例输入1

4 2
1 1 1 1
8 8 1 1
8 8 9 9
1 1 9 9

样例输出1

68

提示

对于20%的数据,1<=k<=n<=50
对于100%的数据,1<=k<=n<=1000
保证答案不超过2^31-1。

---

题解

前缀和（O（N^2）的代码）

用 $f[i][j]$ 表示以 $(i$+$k$-$1,j$+$k$-$1)$ 为右下角的矩形中取一个 $k$$*$$k$ 的矩形的最大值

因为只需要取出两个 $k$$*$$k$ 的矩形即可，那么我们只要枚举 $i,j$ 算出以 $(i,j)$ 为第二个矩形的左上角的值和+$max(f[i$-$k][n],f[n][j$-$k])$ 的最大值即为答案

---

代码

#include<cstdio>
#include<cstring>
#define LL long long
using namespace std;

int n,k;
LL ans;
LL lr[1005][1005],ud[1005][1005],f[1005][1005],map[1005][1005];

int readln()
{
    int x=0;
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while ('0'<=ch&&ch<='9') x=x*10+ch-48,ch=getchar();
    return x;
}

LL max(LL x,LL y){return x>y?x:y;}

int main()
{
    memset(ud,0,sizeof(ud));
    memset(lr,0,sizeof(lr));
    memset(f,0,sizeof(f));
    n=readln();k=readln();
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            map[i][j]=readln();
            ud[i][j]=ud[i-1][j]+map[i][j];
            lr[i][j]=lr[i][j-1]+map[i][j];
        }
    map[1][1]=0;
    for (int i=1;i<=k;i++) map[1][1]+=lr[i][k];
    n-=k-1;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            if (j==1) {
                if (i==1) continue;
                map[i][j]=map[i-1][j]+lr[i+k-1][k]-lr[i-1][k];
            }
            else map[i][j]=map[i][j-1]+ud[i+k-1][j+k-1]+ud[i-1][j-1]-ud[i-1][j+k-1]-ud[i+k-1][j-1];
        }
    for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) f[i][j]=max(map[i][j],max(f[i-1][j],f[i][j-1]));
    for (int i=1;i<=n;i++) for (int j=1;j<=n;j++) ans=max(ans,map[i][j]+max(f[max(i-k,0)][n],f[n][max(j-k,0)]));
    printf("%lld",ans);
    return 0;
}