99. 激光炸弹

[maninbule]

99. 激光炸弹

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

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

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

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

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

输入格式

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

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

输出格式

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

数据范围

0≤R≤1e9
0<N≤10000,
0≤Xi,Yi≤5000
0≤Wi≤1000

输入样例：

2 1
0 0 1
1 1 1

输出样例：

1

[maninbule]

二维前缀和的模板题

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e6+10;

int N,R;
int arr[5010][5010];
int main(){
    cin>>N>>R;
    int x,y,w;
    for(int i = 0;i<N;i++){
        scanf("%d%d%d",&x,&y,&w);
        x++;y++;
        arr[x][y] = w;
    }
    for(int i = 1;i<=5001;i++){
        for(int j = 1;j<=5001;j++){
            arr[i][j] = arr[i][j] + arr[i][j-1]+arr[i-1][j]-arr[i-1][j-1];
        }
    }
    int res = 0;
    for(int i = R;i<=5001;i++){
        for(int j = R;j<=5001;j++){
            res = max(res,arr[i][j]-arr[i-R][j]-arr[i][j-R]+arr[i-R][j-R]);
        }
    }
    cout<<res<<endl;
    return 0;
}