99. 激光炸弹(二维前缀和)

99. 激光炸弹

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

注意：不同目标可能在同一位置。

现在有一种新型的激光炸弹，可以摧毁一个包含 R×R 个位置的正方形内的所有目标。

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

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

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

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

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

数据范围
0≤R≤109
0<N≤10000,
0≤Xi,Yi≤5000
0≤Wi≤1000
输入样例：
2 1
0 0 1
1 1 1
输出样例：
1

#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define inf 1<<30

using namespace std;
int n,r,x,y,w,g[5005][5005],row,col;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> r;
    row=col=r;
    while(n--)
    {
        cin >> x >> y >> w;
        x++;
        y++;
        g[x][y]+=w;
        row=max(row,x);
        col=max(col,y);
    }
    for(int i=1;i<=row;i++)//前缀和
        for(int j=1;j<=col;j++)
            g[i][j]+=g[i-1][j]+g[i][j-1]-g[i-1][j-1];
    int res=0;
    for(int i=r;i<=row;i++)
        for(int j=r;j<=col;j++)
            res=max(res,g[i][j]-g[i-r][j]-g[i][j-r]+g[i-r][j-r]);
    cout << res << endl;
    return 0;
}