激光炸弹c++

题目

输入样例：

2 1
0 0 1
1 1 1

输出样例：

1

思路

        由题知本题要求某个区间内数的和，联想到二维前缀和。我们可以先使用二维前缀和模板计算各区间的价值。然后枚举以某点为右下角，大小为R*R的正方形价值，取最大值。
        有几点需要注意。一个是本题的下标是从0开始的，而二维前缀和模板从1开始，因此下标要变一下。另一个是炸弹爆炸的范围可能比地图区间大，为了更好计算一个爆炸范围内的总价值，可以把地图“扩大”，扩大的地图目标价值为0。最后一个是题目说的目标是在格点而不是格子内的，炸弹只能“摧毁一个包含R×R个位置的正方形内的所有目标”，边上的不可以；对此我们可以把目标“移到”格子中间，那么一个能摧毁2 x 2范围内的炸弹可以炸掉四个格子里的目标。

代码

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

const int N = 5010;
int a[N][N];

int main()
{
  int n, r, x, y, w;
  cin >> n >> r;
  //xi,yi <= 5000，如果爆炸范围大于5000时，扩大地图也不会增加摧毁的总价值
  r = min (5001, r);
  //用于减少二维前缀和的计算
  int cx = r, cy = r;
  while (n --)
  {
    cin >> x >> y >> w;
    a[++x][++y] += w;
    cx = max(cx, x), cy = max(cy, y);
  }
  
//二维前缀和模板
  for (int i = 1; i <= cx; i ++)
  for (int j = 1; j <= cy; j ++)
  a[i][j] += (a[i][j - 1] + a[i - 1][j] - a[i- 1][j - 1]);
  
  int value = 0;
  for (int i = r; i <= cx; i ++)
  {
    for (int j = r; j <= cy; j ++)
    {
      value = max(value, a[i][j] - a[i - r][j] - a[i][j - r] + a[i - r][j - r]);
    }
  }
  
  cout << value;
  return 0;
}