【ACWing】99. 激光炸弹

题目地址：

https://www.acwing.com/problem/content/description/101/

地图上有$N$个目标，用整数$X_i,Y_i$表示目标在地图上的位置，每个目标都有一个价值$W_i$。注意：不同目标可能在同一位置。现在有一种新型的激光炸弹，可以摧毁一个包含$R\times R$个位置的正方形内的所有目标。激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆炸范围，即那个正方形的边必须和$x$，$y$轴平行。求一颗炸弹最多能炸掉地图上总价值为多少的目标。

输入格式：
第一行输入正整数$N$和$R$，分别代表地图上的目标数目和正方形包含的横纵位置数量，数据用空格隔开。
接下来$N$行，每行输入一组数据，每组数据包括三个整数$X_i,Y_i,W_i$，分别代表目标的$x$坐标，$y$坐标和价值，数据用空格隔开。

输出格式：
输出一个正整数，代表一颗炸弹最多能炸掉地图上目标的总价值数目。数据范围$0\le R\le 10^9$，$0<N\le 10000$,$0\le X_i,Y_i\le 5000$,$0\le W_i\le 1000$。

因为地图最大就是$5001\times 5001$这么大，所以$R$最大只能取$5001$。这个题目即为求某个矩阵的宽$R$的子方阵的最大和。可以先求一下二维前缀和，然后暴力枚举子方阵的右下格子的坐标。代码如下：

#include <iostream>
using namespace std;

const int N = 5010;
int g[N][N];
int m, n;

int main() {
  int N, r;
  scanf("%d%d", &N, &r);
  r = min(r, 5001);
  n = m = r;
  for (int i = 0, x, y, w; i < N; i++) {
    scanf("%d%d%d", &x, &y, &w);
    x++, y++;
    g[x][y] += w;
    n = max(n, x), m = max(m, y);
  }

  for (int i = 1; i <= n; i++)
    for (int j = 1; j <= m; j++)
      g[i][j] += g[i - 1][j] + g[i][j - 1] - g[i - 1][j - 1];

  int res = 0;
  for (int i = r; i <= n; i++)
    for (int j = r; j <= m; j++)
      res = max(res, g[i][j] - g[i - r][j] - g[i][j - r] + g[i - r][j - r]);
  printf("%d\n", res);
}

时空复杂度$O(mn)$，$m,n$为整个地图的长和宽。