《算法竞赛进阶指南》0x03 T1 激光炸弹

题目描述

地图上有 N N N 个目标，用整数 X i , Y i X_i,Y_i Xi​,Yi​ 表示目标在地图上的位置，每个目标都有一个价值 W i W_i Wi​。

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

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

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

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

输入格式

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

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

输出格式

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

数据范围

0 ≤ R ≤ 109 0≤R≤109 0≤R≤109
0 < N ≤ 10000 , 0<N≤10000, 0<N≤10000,
0 ≤ X i , Y i ≤ 5000 0≤X_i,Y_i≤5000 0≤Xi​,Yi​≤5000
0 ≤ W i ≤ 1000 0≤W_i≤1000 0≤Wi​≤1000

输入样例

2 1
0 0 1
1 1 1

输出样例

1

题解

二维前缀和
x和y的数据范围非常小，可以直接两重循环先处理前缀和
再计算每个区域的最大值即可
为了更方便的求前缀和，可以将x和y加上1，避免为0的情况
数据中存在一个位置多个目标的数据，在储存a数组时不能只赋值，应该

code

#include<bits/stdc++.h>
using namespace std;
int a[5010][5010];
int main()
{
	int n,R,ans=-1;
	cin>>n>>R;
	for(int i=1;i<=n;i++)
	{
		int x,y,v;
		cin>>x>>y>>v;
		a[x+1][y+1]+=v;
	}
	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]+a[i][j];
	for(int i=0;i<=5001-R;i++)
		for(int j=0;j<=5001-R;j++)
			ans=max(ans,a[i][j]-a[i+R][j]-a[i][j+R]+a[i+R][j+R]);
	cout<<ans;
	return 0;
}

标签：5000,进阶,T1,int,Wi,目标,炸弹,0x03,数据
来源： https://blog.csdn.net/weixin_55355427/article/details/115186918