計蒜客 打地鼠
阿新 • • 發佈:2018-11-03
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; }