【OJ】BZOJ1218: [HNOI2003]激光炸弹 （前缀和）

https://www.lydsy.com/JudgeOnline/problem.php?id=1218

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

解析

借助二维前缀和可解此题。要注意的是，正方形的边界不计入，而炸弹爆炸的边界不一定要在网格线上，那么可以这样考虑：不再使用目标所在作为网格线交点，而使其成为网格中点，而爆炸边界则构成网格线。例如一次爆炸的边界是(0,0)，(0,1)，(1,1)，(1,0)为定点的正方形，而目标为(0.5,0.5)，这样，之后就可以很方便地思考，不用纠结边界问题。不过，初始化的时候原本(0,0)的目标，在新网格中变为(0.5,0.5)其值应放到s[1, 1]上。

Code

#include <iostream>
#include <climits>

using namespace std;

const int N = 10002;
const int R = 5002;

int s[R][R] = {0};

int main()
{
    int n, r, xi, yi, vi;
    int min_xi = INT_MAX, min_yi = INT_MAX, max_xi = 0, max_yi = 0;
    cin >> n >> r;
    for(register int i = 0; i < n; i++)
    {
        cin >> xi >> yi >> vi;
        //注意初始化这里横纵坐标要加1
        s[xi+1][yi+1] += vi;
        min_xi = min(min_xi, xi+1);
        min_yi = min(min_yi, yi+1);
        max_xi = max(max_xi, xi+1);
        max_yi = max(max_yi, yi+1);
    }
    for(register int i = min_xi; i <= max_xi; i++)
    {
        for(register int j = min_yi; j < max_yi; j++)
        {
            s[i][j] += s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
        }
    }
    int ans = 0, i_= 0,j_ = 0;
    for(register int i = min_xi; i <= max_xi; i++)
    {
        for(register int j = min_yi; j <= max_yi; j++)
        {
            i_ = (i-r>0)? i-r : 0;
            j_ = (j-r>0)? j-r : 0;
            ans = max(ans, s[i][j] - s[i_][j] - s[i][j_] + s[i_][j_]);
        }
    }
    cout << ans << endl;
    return 0;
}