AcWing 99.激光炸弹（矩阵前缀和）

整理的算法模板：ACM算法模板总结（分类详细版）

 

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

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

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

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

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

输入格式

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

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

输出格式

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

数据范围

0≤R≤1090≤R≤109
0<N≤100000<N≤10000,
0≤Xi,Yi≤50000≤Xi,Yi≤5000
0≤Wi≤10000≤Wi≤1000

输入样例：

2 1
0 0 1
1 1 1

输出样例：

1

难度：简单

时/空限制：10s / 168MB

总通过数：2577

总尝试数：7169

来源：《算法竞赛进阶指南》

算法标签

 思路：

子矩阵的前缀和

由于每个炸弹的爆炸范围是RxR的正方形内部，边上的不能炸到，所以能炸到的点为(R-1)x(R-1) 个；求完矩阵前缀和，然后枚举每一个(R-1)x(R-1)矩阵的和，然后取最大值即可；

#include <bits/stdc++.h>
using namespace std;
const int N=5005;
int s[N][N];
int main()
{
    int n,r;
    cin >>n>>r;
    r=min(r,5002);
    for(int i=0;i<n;i++)
    {
        int x,y,w;
        cin >>x>>y>>w;
        s[++x][++y]+=w;
    }
    for(int i=1;i<5001;i++)
        for(int j=1;j<5001;j++) s[i][j]+=s[i-1][j]+s[i][j-1]-s[i-1][j-1];
    int ans=-1;
    for(int i=r;i<5001;i++)
        for(int j=r;j<5001;j++)
            ans=max(ans,s[i][j]-s[i-r][j]-s[i][j-r]+s[i-r][j-r]);
    cout <<ans<<endl;
}