牛客练习赛74 C. CCA的矩阵

链接

https://ac.nowcoder.com/acm/contest/9700/C

题意

一个 $n\ast n$ 的矩形内布满了老鼠。

万幸的是，你有一个 $k\ast k$ 的锤子，一锤子砸下去可以把它覆盖到的所有老鼠清除。

这个锤子只能斜着锤，对于一个 $3\ast 3$ 的锤子，它能覆盖到的区域如下：

- - * - -
- * * * -
* * * * *
- * * * -
- - * - -

一锤子砸下去，最多能清除多少只老鼠？

思路

将数组旋转 $45^{\circ }$，求一下前缀和即可。

代码

#include <bits/stdc++.h>
#define SZ(x) (int)(x).size()
#define ALL(x) (x).begin(),(x).end()
#define PB push_back
#define EB emplace_back
#define MP make_pair
#define FI first
#define SE second
using namespace std;
typedef double DB;
typedef long long LL;
typedef pair<int,int> PII;
typedef vector<int> VI;
typedef vector<PII> VPII;
//head
const int N=5e3+5;
int n,k;
LL a[N][N],s[N][N];
int main() {
    //freopen("E:/OneDrive/IO/in.txt","r",stdin);
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cin>>n>>k;
    for(int i=1;i<=n;i++) {
        for(int j=1;j<=n;j++) {
            cin>>a[i+j][i-j+n];
        }
    }
    for(int i=1;i<=n+n+k+k;i++) {
        for(int j=1;j<=n+n+k+k;j++) {
            s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];
        }
    }
    LL res=0;
    for(int i=1;i<=n+n;i++) {
        int a=i+2*(k-1);
        for(int j=1;j<=n+n;j++) {
            if((i+j-n)&1) continue;
            int b=j+2*(k-1);
            res=max(res,s[a][b]-s[i-1][b]-s[a][j-1]+s[i-1][j-1]);
        }
    }
    cout<<res<<'\n';
    return 0;
}