BZOJ1248 激光炸弹 - 前缀和

Description

一种新型的激光炸弹，可以摧毁一个边长为R的正方形内的所有的目标。现在地图上有n个目标，用整数xi,yi(0<=x
i,yi<=5000)表示目标在地图上的位置，每个目标都有一个价值0<vi<100。激光炸弹的投放是通过卫星定位的，但
其有一个缺点，就是其爆破范围，即那个边长为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

思路：二维前缀和s[i][j]=s[i-r][j]+s[i][j-r]-s[i-r][j-r]+a[i][j]，之后找到最大的前缀和。
由于本题空间限制较小，可以省去a数组，直接把价值v加到s[i][j]里。
AC Code：

#include<cstdio>
#include<algorithm>
using namespace std;
const int sz=5000+10;
int s[sz][sz];
int main()
{
    int n,r;
    scanf("%d%d",&n,&r);
    int p=r,q=r;
    while(n--)
    {
        int x,y,w;
        scanf("%d%d%d",&x,&y,&w);
        x++,y++;
        s[x][y]+=w;
        p=max(p,x),q=max(q,y);
    }
    for(int i=1;i<=p;i++)
        for(int j=1;j<=q;j++)
            s[i][j]+=s[i-1][j]+s[i][j-1]-s[i-1][j-1];
    int ans=0;
    for(int i=r;i<=p;i++)
        for(int j=r;j<=q;j++)
            ans=max(ans,s[i][j]-s[i-r][j]-s[i][j-r]+s[i-r][j-r]);
    printf("%d",ans);
    return 0;
}