【洛谷】激光炸弹

[HNOI2003]激光炸弹

题目描述

一种新型的激光炸弹，可以摧毁一个边长为 $m$ 的正方形内的所有目标。现在地图上有 $n$ 个目标，用整数 $x_i$ , $y_i$ 表示目标在地图上的位置，每个目标都有一个价值 $v_i$ .激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆破范围，即那个边长为 $m$ 的边必须与 $x$ 轴, $y$ 轴平行。若目标位于爆破正方形的边上，该目标不会被摧毁。

现在你的任务是计算一颗炸弹最多能炸掉地图上总价值为多少的目标。

输入格式

输入的第一行为整数 $n$ 和整数 $m$，

接下来的 $n$ 行，每行有 $3$ 个整数 $x,y,v$，表示一个目标的坐标与价值。

输出格式

输出仅有一个正整数，表示一颗炸弹最多能炸掉地图上总价值为多少的目标（结果不会超过 $32767$ ）。

样例 #1

样例输入 #1

2 1
0 0 1
1 1 1

样例输出 #1

1

提示

数据规模与约定

• 对于 $100\%$ 的数据，保证 $1\le n\le 10^4$，$0\le x_i,y_i\le 5\times 10^3$，$1\le m\le 5\times 10^3$，$1\le v_i<100$。

代码

#include<iostream>
using namespace std;
int n, m;
const int N = 5010;
int s[N][N];
int get(int x1, int y1, int x2, int y2) {
	return s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1];
}
int main() {
	cin >> n >> m;
	for (int i = 0; i < n; i++) {
		int x, y, v;
		cin >> x >> y >> v;
		x++; y++;
		s[x][y] = max(s[x][y], v);
	}
	for (int i = 1; i <= 5005; i++) {
		for (int j = 1; j <= 5005; j++) {
			s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + s[i][j];
		}
	}
	int ans = 0;
	for (int i = 1; i + m - 1 <= 5002; i++) {
		for (int j = 1; j + m - 1 <= 5002; j++) {
			ans = max(ans, get(i, j, i + m - 1, j + m - 1));
		}
	}
	cout << ans << endl;
	return 0;
}

知识点

二维前缀和 枚举

注意

空间限制125MB 一个二维数组5000*5000*4 =100MB 所以只能开一个。
和模板题不太一样的地方就是目标在格点上