激光炸弹 HYSBZ - 1218 （二维前缀和）

版权声明： https://blog.csdn.net/nucleare/article/details/89432799

激光炸弹

 HYSBZ - 1218 

一种新型的激光炸弹，可以摧毁一个边长为R的正方形内的所有的目标。现在地图上有n(N<=10000)个目标，用整数Xi,Yi(其值在[0,5000])表示目标在地图上的位置，每个目标都有一个价值。激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆破范围，即那个边长为R的正方形的边必须和x，y轴平行。若目标位于爆破正方形的边上，该目标将不会被摧毁。 

Input

输入文件的第一行为正整数n和正整数R，接下来的n行每行有3个正整数，分别表示xi,yi,vi

Output

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

Sample Input

2 1 0 0 1 1 1 1

Sample Output

1

Hint

思路：二维前缀和，枚举右下角

#include <bits/stdc++.h>
using namespace std;

int a[5002][5002];
int main() {
	
	int n, r;
	while (~scanf ("%d %d", &n, &r)) {
		int x, y, w;
		memset(a, 0, sizeof(a));
		for (int i = 1; i <= n; ++i) {
			scanf ("%d %d %d", &x, &y, &w);
			a[x+1][y+1] = w;
		}
		for (int i = 1; i <= 5001; ++i) {
			for (int j = 1; j <= 5001; ++j) {
				a[i][j] += (a[i-1][j] + a[i][j-1] - a[i-1][j-1]);
			}
		}
		int ans = 0;
		for (int i = r; i <= 5001; ++i) {
			for (int j = r; j <= 5001; ++j) {
				ans = max(ans, (a[i][j] - a[i-r][j] - a[i][j-r] + a[i-r][j-r]));
			}
		}
		printf ("%d\n", ans);
	}
	
	return 0;
}