BZOJ 1218 激光炸弹

一、内容

一种新型的激光炸弹，可以摧毁一个边长为 R的正方形内的所有的目标。现在地图上有 N个目标，用整数Xi,Yi表示目标在地图上的位置，每个目标都有一个价值Wi。激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆炸范围，即那个边长为 R的正方形的边必须和x，y平行。若目标位于爆破正方形的边上，该目标不会被摧毁。求一颗炸弹最多能炸掉地图上总价值为多少的目标。
输入格式

第一行输入正整数 N
和 R

,分别代表地图上的目标数目和正方形的边长，数据用空格隔开。

接下来N
行，每行输入一组数据，每组数据包括三个整数Xi,Yi,Wi,分别代表目标的x坐标，y

坐标和价值，数据用空格隔开。
输出格式

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

0<N≤10000
,
0≤X_i,Y_i≤5000

输入样例：

2 1
0 0 1
1 1 1

输出样例：

1

二、思路

• 先求出一个二维前缀和，然后枚举一下R长的正方形，找出最大的答案。

三、代码

#include <cstdio>
#include <iostream>
using namespace std;
#define max(a,b) (a > b ? a : b)
typedef long long ll;
const int N = 5001;

int sum[N + 1][N + 1], n, r, x, y, w , col, row;
ll ans;
void init() {
	//求二维和
	for (int i = 1; i <= row; i++) {
		for (int j = 1; j <= col; j++) {
			sum[i][j] += sum[i][j - 1] + sum[i - 1][j] - sum[i - 1][j - 1];
			printf("%d ", sum[i][j]);
		}
		printf("\n");
	} 
} 
int main() {
	scanf("%d%d", &n, &r);
	col = row = r;
	for (int i = 0; i < n; i++) {
		scanf("%d%d%d", &x, &y, &w);
		x++; y++;
		sum[x][y] += w;
		row = max(row, x);
		col = max(col, y); 
		 
	} 
	init();
	//枚举R的长度
	for (int i = r; i <= row; i++) {
		for (int j = r; j <= col; j++) {
			//我们把点当做每个方格的中心  这时候求的R边长实际是R-1边长 因为不包括边上的点 
			ans = max(ans, sum[i][j] - sum[i][j - r] - sum[i - r][j] + sum[i - r][j - r]);
		}
	} 
	printf("%d", ans);
	return 0;
}