1. 程式人生 > >計蒜客 打地鼠

計蒜客 打地鼠

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