洛谷P2280 [HNOI2003] 激光炸弹

[HNOI2003] 激光炸弹

题目描述

一种新型的激光炸弹，可以摧毁一个边长为 $m$ 的正方形内的所有目标。现在地图上有 $n$ 个目标，用整数 $x_i$ , $y_i$ 表示目标在地图上的位置，每个目标都有一个价值 $v_i$。激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆破范围，即那个边长为 $m$ 的边必须与 $x$ 轴，$y$ 轴平行。若目标位于爆破正方形的边上，该目标不会被摧毁。

现在你的任务是计算一颗炸弹最多能炸掉地图上总价值为多少的目标。

可能存在多个目标在同一位置上的情况。

输入格式

输入的第一行为整数 $n$ 和整数 $m$；

接下来的 $n$ 行，每行有 $3$ 个整数 $x,y,v$，表示一个目标的坐标与价值。

输出格式

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

样例 #1

样例输入 #1

2 1
0 0 1
1 1 1

样例输出 #1

1

提示

数据规模与约定

• 对于 $100\%$ 的数据，保证 $1\le n\le 10^4$，$0\le x_i,y_i\le 5\times 10^3$，$1\le m\le 5\times 10^3$，$1\le v_i<100$。

这道题做法很多什么dp，前缀和等等等等。
今天主讲前缀和。这题有个坑点，就是可能出现m>n的情况，这时候很多题解都选择做特判，但奈何我特判似乎没写对，死磕了半天写出不需要特判的正解代码。思路见代码

Code

#include<bits/stdc++.h>
using namespace std;
int ans=-2000000000,maxn=-1,maxm=-1;//预处理 
int s[5005][5005];//储存地图 
int main(){
	int n,m,x,y,v;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>x>>y>>v;
		s[x+1][y+1]+=v;
		maxn=max(maxn,x+1);
		maxm=max(maxm,y+1);//构建地图 
	}
	for(int i=1;i<=maxn;i++){
		for (int j=1;j<=maxm;j++){
			s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+s[i][j];//二位前缀和处理 
		}
	}
//    if(m >= maxn || m >= maxm){//  maxn < m <maxm 
//    	/*
//			1 1 1 1 1
//			1 1 1 1 1 
//		  	1 1 1 1 1  m = 4 
//		*/
    	assert(0);
//        cout << s[maxn][maxm] << endl;
//        return 0;
//    }
	for(int i=min(m, maxn);i<=maxn;i++){
		for(int j=min(m,maxm);j<=maxm;j++){
			int k=s[i][j]-s[i][max(0,j-m)]-s[max(0,i-m)][j]+s[max(0,i-m)][max(0,j-m)];//处理答案 
			ans=max(ans,k);//判断+储存答案 
		}
	}
	cout<<ans<<endl;//输出 
	return 0;
}