链接:登录—专业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
2 1 0 0 1 1 1 1
输出
复制1
1
备注:
对于100%的数据,保证 1≤n≤104,0≤xi,yi≤5×103,1≤m≤5×103,1≤vi<100。
思路:
二维前缀和差分
代码:
#include<bits/stdc++.h>
using namespace std;
int n,r;
int d[5005][5005];
int main(){
cin.tie(0);
cin>>n>>r;
//r=min(r,5001);
while(n--){
int x,y,v;cin>>x>>y>>v;
d[++x][++y]+=v;
}
for(int i=1;i<=5001;++i){
for(int j=1;j<=5001;++j){
d[i][j]+=d[i-1][j]+d[i][j-1]-d[i-1][j-1];
//cout<<a[i][j]<<" ";
}
//cout<<endl;
}
int res=0;
for(int i=r;i<=5001;++i){
for(int j=r;j<=5001;++j){
res=max(res,d[i][j]-d[i-r][j]-d[i][j-r]+d[i-r][j-r]);
}
// cout<<res<<endl;;
}
cout<<res;
}