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;
}