99.激光炸弹 | 前缀和

 

99. 激光炸弹 - AcWing题库

 

 

// 二维前缀和
#include <iostream>
#include <cmath>
using namespace std;
const int N = 5e3 + 10;
int n, a[N][N], r, SUM = 0;
int main()
{
    cin >> n >> r;
    int x, y, w;
    int x_max = r, y_max = r; // 必须初始化为r 若初始化为0，则建立前缀和数组时无法进入循环(line:36)
    for (int i = 0; i < n; i++)
    {
        scanf("%d%d%d", &x, &y, &w);
        x++, y++; // 激光边界的正方形不会爆炸
        a[x][y] += w;
        SUM += w;//(line:24)
        x_max = max(x_max, x);
        y_max = max(y_max, y);
    }

    if (r > 5000)//数据加强过，需要加个特判：当激光"口径"大于坐标平面范围(0<=x,y<=5000)时，总的毁灭价值为SUM
    {
        cout << SUM << endl;
        return 0;
    }else{
        for (int i = 1; i <= x_max; i++) // 在a数组上直接建立前缀和数组
        {
            for (int j = 1; j <= y_max; j++)
            {
                a[i][j] += a[i - 1][j] + a[i][j - 1] - a[i - 1][j - 1];
            }
        }
    }
    int res = 0;
    for (int i = r; i <= x_max; i++) // 将所有的R*R小方阵遍历一遍 找最大毁灭价值  值得注意的是：i,j都从r开始遍历
    {
        for (int j = r; j <= y_max; j++)
        {
            res = max(res, a[i][j] - a[i][j - r] - a[i - r][j] + a[i - r][j - r]);
        }
    }

    cout << res;

    return 0;
}