蓝桥杯 激光炸弹（二维前缀和）

#include<bits/stdc++.h>
using namespace std;
const int M=5010;
int a[M][M],n,r;
int main(){
    cin>>n>>r;
    r=min(5001,r);
    while(n--){
        int x,y,z;
        cin>>x>>y>>z;
        a[x+1][y+1]+=z;
    }
    for(int i=1;i<=M-1;i++){
        for(int j=1;j<=M-1;j++)a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+a[i][j];
    }
    int res=-1;
    for(int i=r;i<=M-1;i++){
        for(int j=r;j<=M-1;j++){
            res=max(res,a[i][j]-a[i-r][j]-a[i][j-r]+a[i-r][j-r]);
        }
    }
    cout<<res<<endl;
}