计蒜客 打地鼠

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;
}

猜你喜欢

转载自blog.csdn.net/qq_37073764/article/details/82947122
今日推荐