BZOJ 1218: [HNOI2003]激光炸弹 （二维前缀和）

1218: [HNOI2003]激光炸弹

Description

一种新型的激光炸弹，可以摧毁一个边长为R的正方形内的所有的目标。现在地图上有n(N<=10000)个目标，用整数Xi,Yi(其值在[0,5000])表示目标在地图上的位置，每个目标都有一个价值。激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆破范围，即那个边长为R的正方形的边必须和x，y轴平行。若目标位于爆破正方形的边上，该目标将不会被摧毁。 

Input

输入文件的第一行为正整数n和正整数R，接下来的n行每行有3个正整数，分别表示xi,yi,vi

Output

输出文件仅有一个正整数，表示一颗炸弹最多能炸掉地图上总价值为多少的目标（结果不会超过32767）。

Sample Input

2 1

0 0 1

1 1 1

Sample Output

1

 

思路：积一个二维前缀和枚举下矩阵刷新最值就好了......

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
#include<set>
#include<vector>
#include<queue>
#include<stack>
#include<list>
//#include<unordered_map>
using namespace std;
#define ll long long 
const int mod=1e9+7;
const int inf=1e9+7;
 
const int maxn=5e3;
 
int sum[maxn+10][maxn+10]; 
 
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
     
    int n,r;
     
    cin>>n>>r;
    int x,y,w;
     
    int mn=-1;
    for(int i=0;i<n;i++)
    {
        cin>>x>>y>>w;
        x++;
        y++;
        mn=max(mn,x);
        mn=max(mn,y);
        sum[x][y]=w;
    }
     
    for(int i=1;i<=mn;i++)
    {
        for(int j=1;j<=mn;j++)
        {
            sum[i][j]+=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
        }
    }
     
    int maxx=-1;
     
    for(int i=r;i<=mn;i++)
    {
        for(int j=r;j<=mn;j++)
        {
            maxx=max(maxx,sum[i][j]-sum[i-r][j]-sum[i][j-r]+sum[i-r][j-r]);
        }
    }
     
    cout<<maxx<<endl;
     
    return 0;
}