激光炸弹(二维前缀和)——《算法竞赛进阶指南》, HNOI2003

99. 激光炸弹

地图上有 N 个目标,用整数 Xi,Yi 表示目标在地图上的位置,每个目标都有一个价值 Wi。

注意:不同目标可能在同一位置。

现在有一种新型的激光炸弹,可以摧毁一个包含 R×R 个位置的正方形内的所有目标。

激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆炸范围,即那个正方形的边必须和 x,y 轴平行。

求一颗炸弹最多能炸掉地图上总价值为多少的目标。

输入格式
第一行输入正整数 N 和 R,分别代表地图上的目标数目和正方形的边长,数据用空格隔开。

接下来 N 行,每行输入一组数据,每组数据包括三个整数 Xi,Yi,Wi,分别代表目标的 x 坐标,y 坐标和价值,数据用空格隔开。

输出格式
输出一个正整数,代表一颗炸弹最多能炸掉地图上目标的总价值数目。

数据范围
0≤R≤10^9
0<N≤10000,
0≤Xi,Yi≤5000
0≤Wi≤1000
输入样例:
2 1
0 0 1
1 1 1
输出样例:
1

题目思路
二维前缀和
此题如果开两个 s u m [ 5005 ] [ 5005 ] sum[5005][5005] sum[5005][5005]的数组,一个用来存输入数据,一个用来存前缀和空间就超了,所以压缩到一个里面

计算前缀和矩阵
S ( x , y ) = S ( x − 1 , y ) + S ( x , y − 1 ) − S ( x − 1 , y − 1 ) + a [ x ] [ y ] S(x,y) = S(x-1,y)+S(x,y-1)-S(x-1,y-1)+a[x][y] S(x,y)=S(x−1,y)+S(x,y−1)−S(x−1,y−1)+a[x][y];

利用前缀和矩阵,计算一个子矩阵的和
S [ ( x 1 , y 1 ) , ( x 2 , y 2 ) ] = S ( x 2 , y 2 ) − S ( x 2 , y 1 − 1 ) − S ( x 1 − 1 , y 2 ) + S ( x 1 − 1 , y 1 − 1 ) S[(x1,y1),(x2,y2)] = S(x2,y2)-S(x2,y1-1)-S(x1-1,y2)+S(x1-1,y1-1) S[(x1,y1),(x2,y2)]=S(x2,y2)−S(x2,y1−1)−S(x1−1,y2)+S(x1−1,y1−1);

可以自行画图得出公式
这个是求子矩阵和
激光炸弹(二维前缀和)——《算法竞赛进阶指南》, HNOI2003

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 5005;

int sum[5005][5005];
//二维前缀和
int main()
{
    int n,r;
    int s = 0;
    scanf("%d%d", &n, &r);
    
    for(int i=1; i<=n; ++i){
        int x,y,v;
        scanf("%d%d%d",&x,&y,&v);
        sum[x+1][y+1] += v;
        s = max(s,max(x,y));
    }
    s++;
    for(int i=1; i<=s; ++i){
        for(int j=1; j<=s; ++j){
            // cout<<sum[i-1][j]<<' '<<sum[i][j-1]<<' '<<a[i-1][j-1]<<' '<<a[i][j]<<endl;
            sum[i][j] += sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
            // cout<<sum[i][j]<<' ';
        }
        // cout<<endl;
    }
    if(r>5001){
        cout<<sum[s][s]<<endl;
        return 0;
    }
    int res = 0;
    for(int i=1; i<=s-r+1; ++i){
        for(int j=1; j<=s-r+1; ++j){
            int x1 = i,y1 = j,x2 = i+r-1,y2 = j+r-1;
            int t = sum[x2][y2] - sum[x2][y1-1] - sum[x1-1][y2] + sum[x1-1][y1-1];
            res = max(res,t);
        }
    }
    cout<<res<<endl;
    return 0;
}
上一篇:29.字符串与文本类型


下一篇:【HZOJ/Tarjan】#E.采集糖果