今天学习一下二位数组前缀和,签一道水题
问题 G: 激光炸弹
时间限制: 1 Sec 内存限制: 128 MB
题目描述
一种新型的激光炸弹,可以摧毁一个边长为R的正方形内的所有的目标。现在地图上有n(n≤10000)个目标,用整数xi,yi(0≤xi,yi≤5000)表示目标在地图上的位置,每个目标都有一个价值0<vi<100。激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆破范围,即那个边长为R的正方形的边必须和x,y轴平行。若目标位于爆破正方形的边上,该目标将不会被摧毁。
现在你的任务是计算一颗炸弹最多能炸掉地图上总价值为多少的目标。
输入
第一行为正整数n和正整数R,接下来的n行每行有3个正整数,分别表示xi,yi,vi。
输出
仅有一个正整数,表示一颗炸弹最多能炸掉地图上总价值为多少的目标(结果不会超过32767)。
样例输入
2 1
0 0 1
1 1 1
样例输出
1
- 该题目意思为:给你一个之多5000*5000的地图,给你地图上一些点的val,给你一个正方形去框地图上的点,并问框到的点val之和最大是多少
- 用mp[i][j]表示地图上从(1,1)到(i,j)val的和
- 那么如何求得一块特定的正方形内val的和呢?
- 例如上图,我们要求蓝色正方形的val和,那么只需用大正方形-两个红色长方形+减了两边多减了了的绿色正方形
- 即为:mp[i][j]-mp[i-r][j]-mp[i][j-r]+mp[i-r][j-r],r为要求的正方形边长
代码如下
#include <iostream>
#include<bits/stdc++.h>
using namespace std;
int mp[5050][5050];
const int N=5001;
int n,r,x,y,val,maxn;
int main()
{
cin>>n>>r;
for(int i=1;i<=n;i++){
cin>>x>>y>>val;
mp[x+1][y+1]+=val;
}
for(int i=1;i<=N;i++){
for(int j=1;j<=N;j++){
mp[i][j]+=mp[i-1][j];
}
}
for(int i=1;i<=N;i++){
for(int j=1;j<=N;j++){
mp[i][j]+=mp[i][j-1];
}
}
for(int i=r;i<=N;i++){
for(int j=r;j<=N;j++){
maxn=max(maxn,mp[i][j]-mp[i-r][j]-mp[i][j-r]+mp[i-r][j-r]);
}
}
cout<<maxn<<endl;
return 0;
}
另外,注意在两遍循环取求前缀和时,N是5001,如果用了n(输入的正方形大小),会造成RE,原因不明,如果谁知道麻烦评论一下,谢谢