AcWing 99. 激光炸弹 区间前缀和

AcWing 99. 激光炸弹

地图上有 N 个目标，用整数Xi,Yi表示目标在地图上的位置，每个目标都有一个价值Wi。

注意：不同目标可能在同一位置。

现在有一种新型的激光炸弹，可以摧毁一个包含 R×R 个位置的正方形内的所有目标。

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

求一颗炸弹最多能炸掉地图上总价值为多少的目标。

输入格式
第一行输入正整数 N 和 R ,分别代表地图上的目标数目和正方形的边长，数据用空格隔开。

接下来N行，每行输入一组数据，每组数据包括三个整数Xi,Yi,Wi,分别代表目标的x坐标，y坐标和价值，数据用空格隔开。

输出格式
输出一个正整数，代表一颗炸弹最多能炸掉地图上目标的总价值数目。

数据范围
0≤R≤109
0<N≤10000,
0≤Xi,Yi≤5000
0≤Wi≤1000
输入样例：

2 1
0 0 1
1 1 1

输出样例：

1

这道题用到了区间前缀和的做法，先给大家复习一下前缀和的思维,这道题我用区间前缀和的思路，把该范围的价值求出来，然后枚举哪个区间价值最大，不过需要注意样例的取值范围，仔细一点很容易AC。

代码如下：

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 5002 + 10 ;

int n, r;
int s[N][N];

int main()
{
    cin >> n >> r;
    r = min(r, 5001);
    
    for(int i = 0; i < n; i++)
    {   
        int a, b, c;
        cin >> a >> b >> c;
        a ++ , b ++ ;
        s[a][b] += c;
    }
    
    for(int i = 1; i <= 5001; i++)
        for(int j = 1; j <= 5001; j++)
            s[i][j] = s[i][j] + s[i][j - 1] + s[i - 1][j] - s[i - 1][j - 1];
            
    int maxs = 0; 
    for(int i = r; i <= 5001; i++)
        for(int j = r ; j <= 5001; j++)
            maxs = max(maxs, s[i][j] - s[i - r ][j] - s[i][j - r] + s[i - r ][j - r]);
            
    cout << maxs << endl;
    
    return 0;
}