来源:牛客网
1、题目描述
链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
题干
一种新型的激光炸弹,可以摧毁一个边长为R的正方形内的所有的目标。
现在地图上有n(N ≤ 10000)个目标,用整数Xi,Yi(其值在[0,5000])表示目标在地图上的位置,每个目标都有一个价值。
激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆破范围,即那个边长为R的正方形的边必须和x,y轴平行。
若目标位于爆破正方形的边上,该目标将不会被摧毁。
输入描述:
输入文件的第一行为正整数n和正整数R,接下来的n行每行有3个正整数,分别表示 xi,yi ,vi 。
输出描述:
输出文件仅有一个正整数,表示一颗炸弹最多能炸掉地图上总价值为多少的目标(结果不会超过32767)。
示例1
输入
2 1
0 0 1
1 1 1
输出
1
备注:
2、思路
题目让我们选出所给出地图上以R为边长的固定方形区域内的价值之和,难点在于如何在复杂度之内统计出固定区域内的价值之和。爆炸范围可以看做固定方形框,由于要统计整个二维地图所有方形框内的所有可能,为了减少不必要的计算,所以我们要利用前面统计的结果,再结合到固定范围遍历统计,这里采用前缀和的思想。
一维前缀和
如上图所示,m 即为 m + n 位置的前缀和。一维前缀和可以 本位数+前一位数 来实现对一维数组的前n项和计算时借用已经计算的结果使复杂度降到了 O(n)。二维也是类似的,核心是利用已得到的结果来减少计算次数(也是dp思想的一部分)。
二维前缀和
如上图,若我们定义数组一个坐标的左上坐标连同它自身位置数值之和为此位置总合。 p 为白框内所有数值和,即此位置总合 , m 为黑框所有数值和,n 为红框内所有数值和, 我们要计算 x 左上蓝色区域所有数值之和,就可以通过 x = m + n - p 的方式来实现对x位置总合的计算(p重复减去两次),n、m 可算作 x 前缀之一。
那么此题中我们要统计数组内 固定边长为 r 的移动正方形框内的数值总合,可以先通过遍历一次二维数组把所有位置的总合计算出来,作为之后计算的准备。因此我们在处理题目所给坐标时要加 1 ,防止统计第一行和第一列的时候破坏了统计规则的一致性(即 左 +上 - 左上)。
如上图,若 r = 3,我们要计算 x(i,j) 作为右下角的方形框内的数值总合(爆炸范围价值和),原理与上面相同,即 x 的总合 - n的总合 - m的总合 + p的总合
((i , j)左上 - (i , j - r)左上 - (i - r , j)左上 + (i - r , j - r)左上)。
若从 v(r , r) 位置开始遍历,即为以下代码(第一行、第一列都是 0 嘛~~)。
代码:
#include<bits/stdc++.h>
using namespace std;
int map1[5010][5010];// x,y<=5000,开充足的空间
int main()
{
int i, j;
int n, r;
cin >> n >> r;
int maxer = 0;
for (i = 1; i <= n; i++)
{
int x, y, v;
cin >> x >> y >> v;
map1[++x][++y] = v;//题目坐标可能从0开始,整体加1, 为减少判定,满足初始前缀和计算初始状态
maxer = max(maxer, max(x, y));//取出坐标最大值,即矩阵边界下标
}
for (i = 1; i <= maxer; i++)
{
for (j = 1; j <= maxer; j++)
{
map1[i][j] = map1[i - 1][j] + map1[i][j - 1] - map1[i - 1][j - 1] + map1[i][j];//遍历一遍二维数组得到二维前缀和
}
}
int result = 0;//累加变量设初值为 0
for (i = r; i <= maxer; i++)
{
for (j = r; j <= maxer; j++)
{
int tool = 0;
tool += map1[i][j] - map1[i - r][j] - map1[i][j - r] + map1[i - r][j - r];//根据计算出的前缀和,计算出矩阵中爆炸范围方框内的和的最大值
result = max(tool, result);
}
}
cout << result;
return 0;
}
思考:当需要进行固定区域取值并且进行遍历时,可以考虑使用前缀和。