激光炸弹——二维前缀和

题目描述

一种新型的激光炸弹，可以摧毁一个边长为R的正方形内的所有的目标。

现在地图上有n(N ≤ 10000)个目标，用整数Xi,Yi(其值在[0,5000])表示目标在地图上的位置，每个目标都有一个价值。

激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆破范围，即那个边长为R的正方形的边必须和x，y轴平行。

若目标位于爆破正方形的边上，该目标将不会被摧毁。

输入描述

输入文件的第一行为正整数n和正整数R，接下来的n行每行有3个正整数，分别表示 xi，yi ，vi 。

输出描述

输出文件仅有一个正整数，表示一颗炸弹最多能炸掉地图上总价值为多少的目标（结果不会超过32767）。

示例1

输入

2 1

0 0 1

1 1 1

输出

1

思路：

• 定义一个二维矩阵，所有目标都处在这个矩阵中，每个位置表示的是当前目标的价值
• 有价值的目标更新到二维矩阵中
• 将二维矩阵进行前缀和操作， 表示以当前位置为右下角，初始位置为左上角的子矩阵的所有价值之和
• 遍历每一个边长为r的子矩阵，利用前缀和的性质，求出这个子矩阵中的总价值， 维护一个最大的价值

代码实现

#include <iostream>

using namespace std;

const int N = 5010;

int ans[N][N];
int n, r;
int x, y, v;

int main()
{
    scanf("%d%d", &n, &r);
    while(n--)
    {
        scanf("%d%d%d", &x, &y, &v);
        ans[x + 1][y + 1] = v;	// 将当前位置的价值更新到二维矩阵中
    }
    for(int i = 1;i < N; i++)
    {
        for(int j = 1; j < N; j++)
        {
            ans[i][j] = ans[i][j-1] + ans[i-1][j] - ans[i-1][j-1] + ans[i][j];	// 进行前缀和操作
        }
    }
    int res = 0;
    for(int i = 0;i + r< N; i++)	// 以r为边长的子矩阵的左上角坐标
    {
        for(int j = 0; j + r< N; j++)	// 以r为边长的子矩阵的右下角坐标
        {
            // 当前以r为边长的子矩阵的总价值，维护子矩阵的最大总价值
            res = max(res, ans[i + r][j + r] - ans[i + r][j] - ans[i][j + r] + ans[i][j]);
        }
    }
    
    printf("%d\n", res);
    
    return 0;
}