二维前缀和之激光炸弹

原题链接:https://www.acwing.com/problem/content/description/101/


地图上有\(N\)个目标点,用整数\(X_i,Y_i\)表示目标在地图上的位置,每个目标都有一个价值\(W_i\)

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

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

激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆炸范围,即那个正方形的边必须和\(x,y\)轴平行。

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

输入格式
第一行输入正整数\(N\)\(R\),分别代表地图上的目标数目和正方形包含的横纵位置数量,数据用空格隔开。

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

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

数据范围
\(0≤R≤10^9\)
\(0<N≤10000\),
\(0≤X_i,Y_i≤5000\)
\(0≤W_i≤1000\)
输入样例:
2 1
0 0 1
1 1 1
输出样例:
1
求某点的前缀和:
image

s[i][j]=g[i-1][j]+g[i][j-1]-g[i-1][j-1]+g[i][j];

二维前缀和:
image
红色部分的面积就是全部有色的面积减去绿色的面积加上黄色的面积(因为多剪了)
AC代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn=5050;
int n1,r;
int g[maxn][maxn];
int main(){
    cin>>n1>>r;
    int x,y,w;
    int n=r,m=r;
    for(int i=0;i<n1;i++){
        cin>>x>>y>>w;
        x++,y++;
        n=max(x,n);
        m=max(y,m);
        g[x][y]=w;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            g[i][j]=g[i-1][j]+g[i][j-1]-g[i-1][j-1]+g[i][j];            
        }
    }
    int ans=0;
    int t;
    for(int i=r;i<=n;i++){
        for(int j=r;j<=m;j++){
            t=g[i][j]-g[i-r][j]-g[i][j-r]+g[i-r][j-r];//算的是边长是(r-1)的正方形的和,因为边上不加
            ans=max(ans,t);
        }
    }
    cout<<ans<<endl;
}
posted @ 2020-01-25 21:44  哎呦哎(iui)  阅读(141)  评论(0编辑  收藏  举报