题目:输入N个宝物的位置和价值,给出一个炸弹的炸弹范围(是一个正方形)的边长,找出一颗炸弹最多能炸掉的最大价值
题目中的宝物放在一个(x,y)的点上,而我们的炸弹是炸一个(r,r)的范围,所以如果一个点恰好在正方形的边上,是不会被摧毁的,上图炸掉的宝物数量为0
如图,当把这个炸弹范围拖动的时候,一个点正好就被放在了炸弹区域内,说明这个宝物可以被炸掉,所以不妨把r*r矩阵想成(r-1)(r-1)
我们可以运用前缀和的思想:快速求(x1~x2,y1~y2)的和来做这一道题,一次求和的时间复杂度是O(1)的,而我们总共有5000x5000个点,是可以满足条件的
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 5005;
int s[N][N];
/*思路:1.输入(0~5001,0~5001)的所有宝物,并求他们的二位前缀和矩阵
2.枚举边长为(r-1)(r-1)的所有矩阵,然后找到一个炸弹炸掉的最大宝物价值
*/
int query(int x1, int y1, int x2, int y2)
{
return s[x2][y2] - s[x2][y1 - 1] - s[x1][y2 - 1] + s[x1 - 1][y1 - 1];
}
int main()
{
int n, r;
cin >> n >> r;
r = min(5002, r); //r最大值可能是10^9越界了
for (int i = 0;i < n;i++)
{
int x, y, w;
cin >> x >> y >> w;
s[++x][++y] += w; //因为有可能多个宝物在一个点上,所以要用+=
//二分前缀和要用到x-1的值,所以让x++,不考虑边界问题
}
for(int i=1;i<=5002;i++)
for (int j = 1;j <= 5002;j++)
{
s[i][j] += s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
//将原来的数组转化为前缀和数组
}
int res = 0;
//x++后,取值[1,5001]
for(int i=r;i<=5001;i++)
for (int j = r;j <= 5001;j++)
{//i,j表示的是矩形的右下角,i-r,j-r表示的是左上角
//由于是(r-1)(r-1)的正方形,所以左上角顶点(j-(r-1)-1)=(j-r)
res = max(res, s[i][j] - s[i - r][j] - s[i][j - r] + s[i - r][j - r]);
//等于res=max(res,query(i-r+1,j-r+1,i,j));
}
cout << res;
}