题目描述
​ 一种新型的激光炸弹,可以摧毁一个边长为 R 的正方形内的所有的目标。现在地图上有 N(N≤10000) 个目标,用整数 Xi,Yi(其值在[0,5000]) 表示目标在地图上的位置,每个目标都有一个价值 Wi。

激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆破范围,即那个边长为 R 的正方形的边必须和 x,y 轴平行。若目标位于爆破正方形的边上,该目标将不会被摧毁。求一颗炸弹最多能炸掉地图上总价值为多少的目标。

输入
​ 输入的第一行为正整数 N 和正整数 R。(1≤N≤10000)
​ 接下来的 N 行每行有 3 个正整数,分别表示 Xi,Yi,Wi。

输出
​ 输出一颗炸弹最多能炸掉地图上总价值为多少的目标。(结果不会超过 32767)

样例输入
2 1
0 0 1
1 1 1
样例输出
1
数据规模与约定
​ 时间限制:1 s

内存限制:256 M

100% 的数据保证 1≤N≤10000

#include <iostream>
using namespace std;

int G[5005][5005];
int M[5005][5005];

int main() {
int n, r;
cin >> n >> r;
int max_x = 0, max_y = 0;
for (int i = 1; i <= n; i++) {
int x, y, w;
cin >> x >> y >> w;
x++, y++;
max_x = max(x, max_x);
max_y = max(y, max_y);
M[x][y] = max(w, M[x][y]);
}

for (int i = 1; i <= max_x; i++) {
for (int j = 1; j <= max_y; j++) {
G[i][j] = G[i - 1][j] + G[i][j - 1] - G[i - 1][j - 1] + M[i][j];
}
}

int ans = 0;
for (int i = 1; i <= max_x; i++) {//正方形左上角坐标
for (int j = 1; j <= max_y; j++) {
int x = min(i + r-1,max_x);//正方形右下角坐标且不超过边界
int y = min(j + r-1,max_y);
ans = max(ans, G[x][y] - G[i - 1][y] - G[x][j - 1] + G[i - 1][j - 1]);
}
}
cout << ans << endl;
return 0;
}