激光炸弹（二维前缀和）

题目

 大致题意为，一个矩阵内的一些点有价值（另一些点也可以看作有价值，但价值为0），然后你可以用一个边长为m的正方形框住一些点，记这些点的总价值为sum的话，求最大的sum。

思路

显然我们需要枚举正方形框在矩阵中能放的所有位置，但是如果每枚举一个位置都要将正方形框内的所有点依次累加，很难不会想像到会发生tle。我们知道，如果我们想要求某些区间[L1,R1]，[L2,R2]，[L3,R3]……内的所有值的和，我们可以用一维前缀和来优化。现在我们想要求某些边长为m的正方形的区域S1，S2，S3……内的所有值的和，就可以请出我们的二维前缀和。首先我们创建这个矩阵graph，其中graph[x][y]表示矩阵上(x,y)到(0,0)这个点组成的矩形内所有点的价值之和，然后我们便可以表示出矩阵上(x1,y1)到(x2,y2)组成的矩形内所有点的价值之和（二维前缀和的知识）。最后只需要枚举能在矩阵中做左上角的点(p,q)，就能知道右下角的点为(p+m-1,q+m-1),就能通过二维前缀和得到这个矩形内的所有点的价值之和。

代码

//本题考察了二维前缀和 
#include <iostream>
using namespace std;
const int maxn = 5e3+5;
int graph[maxn][maxn];
int main(){
	int n,m,ans=0;
	cin>>n>>m;
	while(n--){
		int x,y,v;
		cin>>x>>y>>v;
		graph[x][y]=v;
	}
	//计算出每个点与（0，0）这个点组成的矩形内的价值的和 
	for(int i=1;i<maxn;i++){
		graph[0][i]+=graph[0][i-1];
		graph[i][0]+=graph[i-1][0];
	}
	for(int i=1;i<maxn;i++){
		for(int j=1;j<maxn;j++){
			graph[i][j]+=graph[i-1][j]+graph[i][j-1]-graph[i-1][j-1];
		}
	}
	//枚举正方形左上角的点 
	for(int i=0;i<maxn-m+1;i++){
		for(int j=0;j<maxn-m+1;j++){
			if(i!=0&&j!=0)ans=max(ans,graph[i+m-1][j+m-1]-graph[i-1][j+m-1]-graph[i+m-1][j-1]+graph[i-1][j-1]);
			else if(i==0&&j!=0)ans=max(ans,graph[i+m-1][j+m-1]-graph[i+m-1][j-1]);
			else if(j==0&&i!=0)ans=max(ans,graph[i+m-1][j+m-1]-graph[i-1][j+m-1]);
			else if(i==0&&j==0)ans=max(ans,graph[i+m-1][j+m-1]);
		}
	}
	cout<<ans;
}