HYSBZ1218 激光炸弹 (二维前缀和+枚举)

激光炸弹

HYSBZ1218

题目描述

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

Input

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

Output

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

Sample Input

2 1
0 0 1
1 1 1

Sample Output

1

分析：

求出二维前缀和然后枚举右下角的坐标。

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<set>
#include<queue>
#include<stack>
#include<map>
#include<string>
#include<algorithm>
#include<sstream>
#include<memory>
#include<utility>
#include<functional>
#include<iterator>
typedef long long ll;
const int inf=0x3f3f3f3f;
const int inn=0x80808080;
using namespace std;
const int maxm=5005;
int d[maxm][maxm];
int n,r;
int main(){
    while(scanf("%d%d",&n,&r)!=EOF){
        memset(d,0,sizeof d);
        int x,y,v;
        for(int i=1;i<=n;i++){
            scanf("%d%d%d",&x,&y,&v);
            d[x+1][y+1]=v;//因为题目有（0，0）的,+1比较方便
        }
        for(int i=1;i<maxm;i++){
            for(int j=1;j<maxm;j++){
                d[i][j]+=d[i][j-1]+d[i-1][j]-d[i-1][j-1];
            }
        }
        int ans=0;
        for(int i=r;i<maxm;i++){
            for(int j=r;j<maxm;j++){
                ans=max(ans,d[i][j]-d[i][j-r]-d[i-r][j]+d[i-r][j-r]);
            }
        }
        printf("%d",ans);
    }
    return 0;
}