题目描述

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

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

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

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

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

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

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

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

数据范围
0≤R≤10^9
0<N≤10000,
0≤Xi,Yi≤5000
0≤Wi≤1000

样例

输入样例：
2 1
0 0 1
1 1 1
输出样例：
1

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>

using namespace std;

const int N = 5001;
int sign[N+1][N+1]; // 0不用 

int main(){
    int i, j, n, r, x, y, w, t, maxn = 0, ans = 0;
    cin >>n >>r;

    for(i = 0; i < n; i++){
        cin >>x >>y >>w;
        sign[x+1][y+1] += w;
        maxn = max(maxn, max(x, y)+1);
    }

    for(i = 1; i <= maxn; i++){
        for(j = 1; j <= maxn; j++){
            sign[i][j] += sign[i][j-1] + sign[i-1][j] - sign[i-1][j-1];

            t = r;
            if(i < r || j < r) t = min(i, j);
            ans = max(ans, sign[i][j] - sign[i-t][j] - sign[i][j-t] + sign[i-t][j-t]);
        }
    }
    cout <<ans;
    return 0;
}