99. 激光炸弹

目录

99. 激光炸弹

输入格式

输出格式

数据范围

输入样例：

输出样例：

代码：

运行结果：

---

99. 激光炸弹

地图上有 NN 个目标点，用整数 Xi,YiXi,Yi 表示目标在地图上的位置，每个目标都有一个价值 WiWi。

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

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

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

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

输入格式

第一行输入正整数 NN 和 RR，分别代表地图上的目标数目和正方形包含的横纵位置数量，数据用空格隔开。

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

输出格式

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

数据范围

0≤R≤10^9
0<N≤10000
0≤Xi,Yi≤5000
0≤Wi≤1000

输入样例：

2 1
0 0 1
1 1 1

输出样例：

1

难度：简单

时/空限制：1s / 168MB

来源： 《算法竞赛进阶指南》HNOI2003

算法标签 二维前缀和

---

代码：

#include<bits/stdc++.h>

using namespace std;

//const int N=5010;
const int N = 5e3 + 10;

int n,m;  //地图长和宽
int cnt,R;//目标个数 边长

int s[N][N];

int main(){
    cin>>cnt>>R;
    R=min(R,5001);
    n=m=R;
    
    //读入信息
    while(cnt--){
        int x,y,w;
        cin>>x>>y>>w;
        x++;y++; //右移一位, 就不需要考虑边界了
        n=max(n,x); m=max(m,y); 
        s[x][y]+=w; //并且必须是+=, 不能是=, 因为一个位置可能会有多个目标
    }
    
    //预处理前缀和
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            s[i][j]+=s[i-1][j]+s[i][j-1]-s[i-1][j-1];
    
    int res=0;
    //枚举所有边长为R的矩形，枚举（i,j）为右下角
    for(int i=R;i<=n;i++)
        for(int j=R;j<=m;j++)
            res=max(res,s[i][j]-s[i-R][j]-s[i][j-R]+s[i-R][j-R]);
            
    cout<<res<<endl;
    
    return 0;
}

运行结果：