https://nanti.jisuanke.com/t/16445 原题在这
首先将原矩阵顺时针旋转45度,公式为
(i,j)->(i+j-1,n-i+j)
然后执行二维前缀和就可以了
当然要注意旋转后整个矩阵会扩大,所以新矩阵和新锤子都要2*x-1,然后因为矩阵扩大了,有很多点加了进来,所以要判断,新矩阵的点(x,y)是原矩阵的才能敲下去,不然会wa(因为敲出去了)
#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
int n,k,ans;
int mp[5006][5006];
int sum[5006][5006];
bool vis[5006][5006];
int main()
{
cin>>n>>k;
int m=2*n-1;
k=k*2-1;
k=min(k,m);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
int s=i+j-1,t=n-i+j;
scanf("%d",&mp[s][t]);
vis[s][t]=true;
}
}
for(int i=1;i<=m;i++)
{
for(int j=1;j<=m;j++)
{
sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+mp[i][j];
}
}
for(int i=1;i<=m-k;i++)
{
for(int j=1;j<=m-k;j++)
{
if(vis[i][j])
{
int S=sum[i][j]-sum[i+k][j]-sum[i][j+k]+sum[i+k][j+k];
ans=max(ans,S);
}
}
}
cout<<ans;
}