题目链接
题意
一种新型的激光炸弹,可以摧毁一个边长为R的正方形内的所有的目标。
现在地图上有n(N ≤ 10000)个目标,用整数Xi,Yi(其值在[0,5000])表示目标在地图上的位置,每个目标都有一个价值。
激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆破范围,即那个边长为R的正方形的边必须和x,y轴平行。
若目标位于爆破正方形的边上,该目标将不会被摧毁。
思路
- 前缀和,sum[i][j]表示a[0-i][0-j]的和
- 注意数组边界
参考代码
#include<bits/stdc++.h>
using namespace std;
int aa[6010][6010];
int sum[6010][6010];
int main()
{
int n,r;
cin>>n>>r;
int x,y,k;
int l=0,w=0;
for(int i=0; i<n; i++)
{
cin>>x>>y>>k;
l=max(l,x);
w=max(w,y);
aa[x+1][y+1]+=k;
}
sum[0][0]=aa[0][0];
for(int i=1; i<=6000; i++)
{
sum[i][0]=sum[i-1][0]+aa[i][0];
}
for(int i=1; i<=6000; i++)
{
sum[0][i]=sum[0][i-1]+aa[0][i];
}
for(int i=0; i<=6000; i++)
{
for(int j=0; j<=6000; j++)
{
if(i>=1&&j>=1)
sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+aa[i][j];
}
}
int ans=-999999;
for(int i=r; i<=6000; i++)
{
for(int j=r; j<=6000; j++)
{
ans=max(ans,sum[i-r][j-r]-sum[i][j-r]-sum[i-r][j]+sum[i][j]);
}
}
cout<<ans<<endl;
}