激光炸弹 二维前缀和

题目链接

https://www.acwing.com/problem/content/101/

或者 https://vjudge.net/problem/HYSBZ-1218

题目

一种新型的激光炸弹，可以摧毁一个边长为 R 的正方形内的所有的目标。

现在地图上有 N个目标，用整数Xi,Yi表示目标在地图上的位置，每个目标都有一个价值Wi。

激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆炸范围，即那个边长为 RR的正方形的边必须和x，y轴平行。

若目标位于爆破正方形的边上，该目标不会被摧毁。

求一颗炸弹最多能炸掉地图上总价值为多少的目标。

输入格式

第一行输入正整数 N 和 R ,分别代表地图上的目标数目和正方形的边长，数据用空格隔开。

接下来N行，每行输入一组数据，每组数据包括三个整数Xi,Yi,Wi,分别代表目标的x坐标，y坐标和价值，数据用空格隔开。

输出格式

输出一个正整数，代表一颗炸弹最多能炸掉地图上目标的总价值数目。

数据范围

0<N≤10000,
0≤Xi,Yi≤5000

样例

输入

2 1

0 0 1

1 1 1

输出

1

代码

#include<iostream>
using namespace std;
const int maxn=5005;
int g[maxn][maxn];
int main(){
    int t,r;
    cin>>t>>r;
    int n=r,m=r;  //记录行，列最大值
    while(t--){
        int x,y,w;
        cin>>x>>y>>w;
        x++,y++;
        n=max(n,x);
        m=max(y,m);
        g[x][y]+=w;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            g[i][j]+=g[i-1][j]+g[i][j-1]-g[i-1][j-1];
        }
    }
    int ans=0;
    for(int i=r;i<=n;i++){
        for(int j=r;j<=m;j++){
            ans=max(ans,g[i][j]-g[i-r][j]-g[i][j-r]+g[i-r][j-r]);
        }
    }
    cout<<ans<<endl;
    return 0;
}