激光炸弹:二维前缀和

激光炸弹

题目描述:

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

输入格式:

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

输出格式:

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

数据范围:

0<N≤10000
0≤Xi,Yi≤5000

输入样例:

2 1
0 0 1
1 1 1

输出样例:

1

题目大意:求一个矩形内边长为r的正方形内的权值,但是不知道矩形高度宽度,并且计算时正方形边上的不能算。

解题描述:用n,m在输入的时候,记录一下最大高度和宽度,另外一个需要注意的是,矩形开始坐标是从0开始的,咱们构建的是从1开始的,所以输入的时候给+1.

感觉难点就在于边界处理。

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N=5010;
int g[N][N];
int main(){
	int q,r;
	scanf("%d%d",&q,&r);
	int n=r,m=r;
	for(int i=0,x,y,w;i<q;i++){
		scanf("%d%d%d",&x,&y,&w);
		x++;													//咱们开始是从1开始的 ,注意+1 
		y++;
		n=max(n,x);
		m=max(m,y);
		g[x][y]+=w;
	}
	
	for(int x=1;x<=n;x++)
		for(int y=1;y<=m;y++)
			g[x][y]+=g[x-1][y]+g[x][y-1]-g[x-1][y-1];			//求前缀和 
		
	int res=0;
	for(int i=r;i<=n;i++)
	for(int j=r;j<=m;j++)
	res=max(res,g[i][j]-g[i-r][j]-g[i][j-r]+g[i-r][j-r]);		//注意边界不算 !!! 
	
	printf("%d\n",res);
	return 0;
}

猜你喜欢

转载自www.cnblogs.com/52dxer/p/10530308.html