二维前缀和

问题 F: 激光炸弹

时间限制: 1 Sec  内存限制: 128 MB
提交: 59  解决: 15
[提交] [状态] [讨论版] [命题人:admin]

题目描述

一种新型的激光炸弹，可以摧毁一个边长为R的正方形内的所有的目标。现在地图上有n（n≤10000）个目标，用整数xi,yi（0≤xi,yi≤5000）表示目标在地图上的位置，每个目标都有一个价值0＜vi＜100。激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆破范围，即那个边长为R的正方形的边必须和x，y轴平行。若目标位于爆破正方形的边上，该目标将不会被摧毁。

现在你的任务是计算一颗炸弹最多能炸掉地图上总价值为多少的目标。

 

输入

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

输出

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

2 1
0 0 1
1 1 1

样例输出

1

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

int Map[5005][5005];

int main()
{
    int n,r;
    cin>>n>>r;
    memset(Map,0,sizeof(Map));
    for(int i=0;i<n;i++)
    {
        int x,y,v;
        cin>>x>>y>>v;
        Map[x+1][y+1] = v;
    }
    for(int i=1;i<=5001;i++)
    {
        for(int j=1;j<=5001;j++)
        {
            Map[i][j] +=Map[i][j-1];
        }
    }
    for(int i=1;i<=5001;i++)
    {
        for(int j=1;j<=5001;j++)
        {
            Map[i][j] += Map[i-1][j];
        }
    }
    long long ans = -1;
    for(int i=r;i<=5001;i++)
    {
        for(int j=r;j<=5001;j++)
        {
            long long int now = Map[i][j]-Map[i][j-r]-Map[i-r][j]+Map[i-r][j-r];
            ans = max(ans,now);
        }
    }
    cout<<ans;
    return 0;
}

// https://blog.csdn.net/lzr010506/article/details/50552554