二维前缀和-激光炸弹

激光炸弹

https://www.luogu.com.cn/problem/P2280

题意

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

思路

1. 定义一个二维数组
2. 通过循环并存储数组a的对应位置
   3.最后计算出辅助数组a

算法一：二维前缀和

实现步骤

1. 定义一个二维数组，储存地图上每个目标的价值
2. 通过循环读入坐标和价值
3. 通过动态规划计算数组a

代码

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,m;
    cin>>n>>m;
    int a[5010][5010]={0};
    for(int i=0;i<n;i++){
        int x,y,v;
        cin>>x>>y>>v;
        a[x][y]=v;
    }
    int ans=0;
    for(int i=0;i<5010;i++){
        for(int j=0;j<5010;j++){
            if(i!=0){
                a[i][j]+=a[i-1][j];
            }
            if(j!=0){
                a[i][j]+=a[i][j-1];
            }
            if(i!=0&&j!=0){
                a[i][j]-=a[i-1][j-1];
            }
        }
    }
    for(int i=0;i<5010;i++){
        for(int j=0;j<5010;j++){
            int sum=a[i][j];
            if(i-m>=0){
                sum-=a[i-m][j];
            }
            if(j-m>=0){
                sum-=a[i][j-m];
            }
            if(i-m>=0&&j-m>=0){
                sum+=a[i-m][j-m];
            }
            ans=max(ans,sum);
        }
    }
    cout<<ans<<endl;
}