[HNOI2003]激光炸弹(二维前缀和+大坑点)

地图上有 N 个目标,用整数 Xi,Yi表示目标在地图上的位置,每个目标都有一个价值 Wi。

注意:不同目标可能在同一位置。

现在有一种新型的激光炸弹,可以摧毁一个包含 R×R个位置的正方形内的所有目标。

激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆炸范围,即那个正方形的边必须和 x,y 轴平行。若目标位于爆破正方形的边上,该目标将不会被摧毁.

求一颗炸弹最多能炸掉地图上总价值为多少的目标。

输入格式

第一行输入正整数 N 和 R,分别代表地图上的目标数目和正方形的边长,数据用空格隔开。

接下来 N 行,每行输入一组数据,每组数据包括三个整数 ,分别代表目标的 x 坐标,y 坐标和价值,数据用空格隔开。

输出格式

输出一个正整数,代表一颗炸弹最多能炸掉地图上目标的总价值数目。

数据范围

0≤R≤1090≤R≤109
0 0≤Xi,Yi≤50000≤Xi,Yi≤5000
0≤Wi≤10000≤Wi≤1000

输入样例:

2 1
0 0 1
1 1 1

输出样例:

1

 二位前缀和不多BB,重要的是有一个大坑!!!!!!看注释!

#include 
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define mem(a) memset(a, 0, sizeof(a))
const int N = 1e4 + 10;
ll f[5010][5010];
int main()
{
    int n, R;
    cin >> n >> R;
    R = min(5001, R); //这是一个坑点!!,R太大下面数组会超范围,所以要预处理
    //int maxx=0,maxy=0;
    int maxx, maxy;
    maxx = maxy = R;//超级大坑! 你忽略了炸弹范围特别特别大,有权的点范围特比特别小,如果你写成注释那样,就会导致后面的前缀和为0算出来肯定是错的。焯!!!!
    for (int i = 1, x, y, v; i <= n; i++)
    {
        cin >> x >> y >> v;
        x += 1;
        y += 1;
        f[x][y] += v;
        maxx = max(x, maxx), maxy = max(y, maxy);
    }
    // cout << maxx << " " << maxy;
    for (int i = 1; i <= maxx; i++)
    {
        for (int j = 1; j <= maxy; j++)
        {
            f[i][j] += f[i - 1][j] + f[i][j - 1] - f[i - 1][j - 1];
        }
    }
    ll ans = 0;
    for (ll i = R; i <= maxx; i++)
    {
        for (ll j = R; j <= maxy; j++)
        {
            ans = max(ans, f[i][j] - f[i - R][j] - f[i][j - R] + f[i - R][j - R]);
        }
    }
    cout << ans << endl;
}

你可能感兴趣的:(思维题,前缀和,二位前缀和)