激光炸弹
地图上有 N 个目标,用整数Xi,Yi表示目标在地图上的位置,每个目标都有一个价值Wi。
注意:不同目标可能在同一位置。
现在有一种新型的激光炸弹,可以摧毁一个包含 R×R 个位置的正方形内的所有目标。
激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆炸范围,即那个正方形的边必须和x,y轴平行。
求一颗炸弹最多能炸掉地图上总价值为多少的目标。
输入格式
第一行输入正整数 N 和 R ,分别代表地图上的目标数目和正方形的边长,数据用空格隔开。
接下来N行,每行输入一组数据,每组数据包括三个整数Xi,Yi,Wi,分别代表目标的x坐标,y坐标和价值,数据用空格隔开。
输出格式
输出一个正整数,代表一颗炸弹最多能炸掉地图上目标的总价值数目。
数据范围
0≤R≤109
0<N≤10000,
0≤Xi,Yi≤5000
0≤Wi≤1000
输入样例:
2 1
0 0 1
1 1 1
输出样例:
1
把二维前缀和的表达式用dp的方式表示:
g[i][j] += g[i-1][j] + g[i][j-1] - g[i-1][j-1]
把所有边长是R的正方形数组都算一遍,取最大值
#include<iostream>
using namespace std;
const int N = 5010;
int g[N][N];
int main()
{
int n,r;
cin >> n >> r;
int x,y,w;
int q=r,m=r;
for(int i = 0; i < n; i++)
{
cin >> x >> y >> w;
x++;
y++;
q = max(q,x);//长和宽的边界
m = max(m,y);
g[x][y] += w;
}
for(int i = 1; i <= q; i++)
for(int j = 1; j <= m; j++)
g[i][j] += g[i-1][j] + g[i][j-1] - g[i-1][j-1];
int res = 0;
for(int i = r;i<=q;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]);
cout << res << endl;
return 0;
}