P2280 [HNOI2003]激光炸弹-前缀和

#include <bits/stdc++.h>
using namespace std;
#define de(x) cout<<x<<" ";
#define Pu puts("");
#define sf(x) scanf("%d",&x);
typedef long long ll;
const int N=5e3+10,mod=100003;
const int inf=0x3f3f3f3f;
int pre[N][N];//最多只能开一个
int n,m,ans;
int main(){
    cin>>n>>m;
    int x,y,z;
    int l=0,r=0;
    for(int i=1;i<=n;i++){
        sf(x)sf(y)sf(z)
        x+=1;y+=1;//变换一下坐标
        if(x>l) l=x;
        if(y>r) r=y;
        pre[x][y]+=z;//注意有可能同一个地方有两个目标
    }
    l=max(l,m);r=max(r,m);//注意这里
    for(int i=1;i<=l;i++){
        for(int j=1;j<=r;j++){
            pre[i][j]+=pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1];
        }
    }
    //原本想在这里加一个特判的，但是注意有可能l或者r有一个小于m
    int tmp;
    for(int i=m;i<=l;i++){
        for(int j=m;j<=r;j++){
            tmp=pre[i][j]-pre[i-m][j]-pre[i][j-m]+pre[i-m][j-m];
            ans=max(ans,tmp);
        }
    }
    printf("%d\n",ans);
    return 0;
}