AcWing 99. 激光炸弹

这是一个二维前缀和的问题，
首先计算好所有点的前缀和，
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;
}