bzoj 1218 [HNOI2003]激光炸弹 二维前缀和

 [HNOI2003]激光炸弹

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 3022  Solved: 1382
[Submit][Status][Discuss]

Description

一种新型的激光炸弹，可以摧毁一个边长为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

HINT

 

#include<cstring>
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>

#define N 5007
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,r,ans;
int sum[N][N];

int main()
{
    n=read(),r=read();
    for (int i=1;i<=n;i++)
    {
        int x=read(),y=read(),z=read();
        sum[x+1][y+1]+=z;
    }
    for (int i=1;i<=5001;i++)
        for (int j=1;j<=5001;j++)
            sum[i][j]+=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
    for (int i=r;i<=5001;i++)
        for (int j=r;j<=5001;j++)
            ans=max(ans,sum[i][j]+sum[i-r][j-r]-sum[i-r][j]-sum[i][j-r]);
    printf("%d\n",ans);
}