《算法竞赛进阶指南》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;
}
上一篇:MVC5 + EF6 + Bootstrap3 (16) 客户端验证


下一篇:Revers篇:攻防世界reverse进阶re2-cpp-is-awesome