激光炸弹(思维+二维前缀和)

版权声明:本文为博主原创文章,欢迎转载。如有问题,欢迎指正。 https://blog.csdn.net/weixin_42172261/article/details/87618492

一种新型的激光炸弹,可以摧毁一个边长为R的正方形内的所有的目标。现在地图上有n(N<=10000)个目标,用整数Xi,Yi(其值在[0,5000])表示目标在地图上的位置,每个目标都有一个价值。激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆破范围,即那个边长为R的正方形的边必须和x,y轴平行。若目标位于爆破正方形的边上,该目标将不会被摧毁。

Input
输入文件的第一行为正整数n和正整数R,接下来的n行每行有3个正整数,分别表示xi,yi,vi

Output
输出文件仅有一个正整数,表示一颗炸弹最多能炸掉地图上总价值为多少的目标(结果不会超过32767)。

在二维前缀和的基础上,a[i][j]-a[i-r][j]-a[i][j-r]+a[i-r][j-r]来表示以(i, j)为右下角变长为r的正方形的价值。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm> 
using namespace std;
const int N=5001;
int n, r;
int a[5005][5005], ans;
int main()
{
	scanf("%d%d", &n, &r);
	for (int i=1; i<=n; i++){
		int x, y, w;
		scanf("%d%d%d", &x, &y, &w);
		a[x+1][y+1]=w;
	}
	
	for (int i=1; i<=N; i++)
		for (int j=1; j<=N; j++) 
			a[i][j]=a[i][j]+a[i-1][j]+a[i][j-1]-a[i-1][j-1];
	for (int i=r; i<=N; i++)
		for (int j=r; j<=N; j++)
			ans=max(ans, a[i][j]-a[i-r][j]-a[i][j-r]+a[i-r][j-r]);
	printf("%d", ans);
	return 0;
} 

猜你喜欢

转载自blog.csdn.net/weixin_42172261/article/details/87618492