这是一个二维前缀和的问题,
首先计算好所有点的前缀和,
s[i][j]=a[i][j]+s[i-1][j]+s[i][j-1]-s[i-1][j-1]
然后求一块区域的二维前缀和
ans=s[i][j]-s[i-m][j]-s[i][j-m]+s[i-m][j-m]
在通过枚举所有的点求出的ans,求最大值。
特别注意,不能用2个5000*5000的数组,会爆内存
#include <bits/stdc++.h>
using namespace std;
int a[5010][5010];
int ans,r,n,w,m,x,y;
int main(){
cin>>n>>r;
for(int i=1;i<=n;i++){
cin>>x>>y>>w;
x++,y++;
m=max(x,m);
m=max(y,m);//获得x,y的最大值
a[x][y]+=w;
}
if(r>m) r=m;//如果半径的长度超过了m
for(int i=1;i<=m;i++){//求所有点的前缀和
for(int j=1;j<=m;j++){
a[i][j]=a[i][j]+a[i-1][j]+a[i][j-1]-a[i-1][j-1];//此处特别注意,定义两个5000*5000的数组会mle
}
}
for(int i=r;i<=m;i++){//枚举每一个以i,j为结束点的前面r*r区域的值,求最大值
for(int j=r;j<=m;j++){
ans=max(ans,a[i][j]-a[i-r][j]-a[i][j-r]+a[i-r][j-r]);
}
}
cout<<ans;
}