AcWing 99. 激光炸弹

发布时间 2023-03-30 14:50:42作者: 回忆、少年

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

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

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

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

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

输入格式

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

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

输出格式

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

数据范围

0R1e9

0<N10000,
0Xi,Yi5000
0Wi1000

 

输入样例:

2 1
0 0 1
1 1 1

输出样例:

1

代码实现:

#include<iostream>
using namespace std;
const int N=5005;
int a[N][N];

int main(){
	int n,r;
	cin>>n>>r;
	int x,y,z;
	int max1=0,max2=0;
	for(int i=0;i<n;i++){
		cin>>x>>y>>z;
		x++,y++;
		a[x][y]+=z;
		max1=max(max1,x);
		max2=max(max2,y);
	}
	for(int i=1;i<=max1;i++){
		for(int j=1;j<=max2;j++){
			a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1];
		}
	}
	int res=-0x3f3f3f3f;
	for(int i=1;i<=max1;i++){
		for(int j=1;j<=max2;j++){
			int x=min(max1,i+r-1),y=min(max2,j+r-1);
			res=max(res,a[x][y]+a[i-1][j-1]-a[i-1][y]-a[x][j-1]);
		}
	}
	cout<<res<<endl;
	return 0;
}