BZOJ1218: [HNOI2003]激光炸弹（二维前缀和）

1218: [HNOI2003]激光炸弹
Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 6009 Solved: 2359
[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
Source

思路： 就是维护一个二维前缀和就可以了。
二维前缀和的计算类似容斥原理
二维前缀和的计算：f[i][j] = f[i-1][j] + f[i][j-1] - f[i-1][j-1] + a[i][j];
正方形区域的计算：f[i][j] - f[i - r][j] - f[i][j - r] + f[i - r][j -r]
关于原理图画什么的，lyd的《进阶指南》里面讲的很清楚啦

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

int f[5005][5005];
short a[5005][5005];
int n,r;

int main()
{
    scanf("%d%d",&n,&r);
    int row = r, col = r;
    for(int i = 1;i <= n;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        x++;y++;
        a[x][y] = z;
        row = max(row,x);
        col = max(col,y);
    }
    
    for(int i = 1;i <= row;i++)
    {
        for(int j = 1;j <= col;j++)
        {
            f[i][j] = f[i - 1][j] + f[i][j - 1] - f[i - 1][j - 1] + a[i][j];
        }
    }
    
    int ans = 0;
    for(int i = r;i <= row;i++)
    {
        for(int j = r;j <= col;j++)
        {
            ans = max(f[i][j] - f[i - r][j] - f[i][j - r] + f[i - r][j - r],ans);
        }
    }
    
    printf("%d\n",ans);
    return 0;
}