激光炸弹(二维前缀和)

激光炸弹

题目链接
地图上有 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

算法分析

标准的二维前缀和
处理边界的时候改一下就可以了

代码实现

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=5e3+5;
typedef long long ll;
int he[maxn][maxn];
int n,r;
int main()
{
    int a,b;
    cin>>n>>r;
    r=min(r,maxn);//防止r的范围太大
    a=b=r;
    int x,y,w;
    for(int i=1;i<=n;i++)
    {
        cin>>x>>y>>w;
        a=max(a,++x);//处理边界
        b=max(b,++y);//处理边界
        he[x][y]+=w;
    }
    for(int i=1;i<=a;i++)
        for(int j=1;j<=b;j++)
            he[i][j]=he[i-1][j]+he[i][j-1]-he[i-1][j-1]+he[i][j];
    int res=0;
    for(int i=r;i<=a;i++)//枚举正方形的右下角
    {
        for(int j=r;j<=b;j++)
        {
            res=max(res,he[i][j]-he[i][j-r]-he[i-r][j]+he[i-r][j-r]);
        }
    }
    cout<<res<<endl;
    return 0;
}