地图上有 N个目标,用整数 Xi,Yi表示目标在地图上的位置,每个目标都有一个价值 Wi。
注意:不同目标可能在同一位置。
现在有一种新型的激光炸弹,可以摧毁一个包含 R×R 个位置的正方形内的所有目标。
激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆炸范围,即那个正方形的边必须和 x,y 轴平行。
求一颗炸弹最多能炸掉地图上总价值为多少的目标。
输入格式
第一行输入正整数 N 和 R,分别代表地图上的目标数目和正方形包含的横纵位置数量,数据用空格隔开。
接下来 N 行,每行输入一组数据,每组数据包括三个整数 Xi,Yi,Wi,分别代表目标的 x 坐标,y 坐标和价值,数据用空格隔开。
输出格式
输出一个正整数,代表一颗炸弹最多能炸掉地图上目标的总价值数目。
数据范围
0≤R≤109
0<N≤10000,
0≤Xi,Yi≤5000
0≤Wi≤1000
输入样例:
2 1
0 0 1
1 1 1
输出样例:
1
题意:
求范围为边长为r的正方形(不包含边界)的炸弹最多能炸出多少价值
分析:
题目中给出目标的x跟y和w,所以我们可以考虑用二维数组存储这三个变量,a[x][y]=w;
题目要求求出r范围内的最大价值,我们可以想到用二维前缀和来直接求出,但由于r的边界是不能包含的,所以求二维数组的r范围的公式要稍微改变一下,具体可以看代码注释
#include <iostream>
#include<algorithm>
using namespace std;
int a[5005][5005];
int main()
{
int n,r,x,y,w;
cin>>n>>r;
//因为x,y的范围在0~5000,那么5001的爆炸范围就可以炸了全部
r=min(r,5001);
for(int i=1;i<=n;++i){
cin>>x>>y>>w;
//二维前缀和的公式中第0行和第0列都是得为0,不然会出现很多问题,因此我们将x,y整体后移一位
//注意这里是+=,不是=,题目明确说了同个位置可以有多个目标
a[++x][++y]+=w;
}
//求前缀和
for(int i=1;i<=5001;++i){
for(int j=1;j<=5001;++j){
a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1];
}
}
int ans=0;
//枚举边长为r的正方形的右下角坐标,i跟j都从r开始,因为小于r的值只会更小
for(int i=r;i<=5001;++i){
for(int j=r;j<=5001;++j){
ans=max(ans,a[i][j]-a[i-r][j]-a[i][j-r]+a[i-r][j-r]);
}
}
cout<<ans;
return 0;
}