BZOJ 1218 算法竞赛进阶指南 二维前缀和

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

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

input

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

output

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

input

2 1

0 0 1

1 1 1

output

1

#include<iostream>
#include<cstdio> 
#include<queue>
#include<cstring>
#include<map>
using namespace std;
#define ll long long
const int len=5e3+4;
int sum[len][len];
int main()
{
	int n,r;
	cin>>n>>r;
	int u,v,w;
	for(int i=0;i<n;++i)
	{
		scanf("%d%d%d",&u,&v,&w);
		sum[u][v]=w;
	}
	for(int i=0;i<=5000;++i)
		for(int j=0;j<=5000;++j)
		{
			if(i==0&&j==0)continue;
			if(i==0)sum[i][j]+=sum[i][j-1];
			else
			if(j==0)sum[i][j]+=sum[i-1][j];
			else
			sum[i][j]+=sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1];
		}
	int maxm=0;
	int t;
	for(int i=0;i<=5000;++i)
		for(int j=1;j<=5000;++j)
		{
			if(j-r<0&&i-r<0)
				t=sum[i][j];
			else
			if(j-r<0)
				t=sum[i][j]-sum[i-r][j];
			else
			if(i-r<0)
				t=sum[i][j]-sum[i][j-r];
			else
				t=sum[i][j]-sum[i][j-r]-sum[i-r][j]+sum[i-r][j-r];
			maxm=max(maxm,t);
		}
		cout<<maxm<<endl;
}