题目:
地图上有 NN 个目标,用整数 Xi,YiXi,Yi 表示目标在地图上的位置,每个目标都有一个价值 WiWi。
注意:不同目标可能在同一位置。
现在有一种新型的激光炸弹,可以摧毁一个包含 R×RR×R 个位置的正方形内的所有目标。
激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆炸范围,即那个正方形的边必须和 x,yx,y 轴平行。
求一颗炸弹最多能炸掉地图上总价值为多少的目标。
输入格式
第一行输入正整数 NN 和 RR,分别代表地图上的目标数目和正方形的边长,数据用空格隔开。
接下来 NN 行,每行输入一组数据,每组数据包括三个整数 Xi,Yi,WiXi,Yi,Wi,分别代表目标的 xx 坐标,yy 坐标和价值,数据用空格隔开。
输出格式
输出一个正整数,代表一颗炸弹最多能炸掉地图上目标的总价值数目。
数据范围
0≤R≤1090≤R≤109
0<N≤100000<N≤10000,
0≤Xi,Yi≤50000≤Xi,Yi≤5000
0≤Wi≤10000≤Wi≤1000
输入样例:
2 1
0 0 1
1 1 1
输出样例:
1
代码:
#include <iostream>
#include <cstring>
#include<algorithm>
using namespace std;
const int N=5010;
int n,m;
int s[N][N];
int main()
{
int t,R;
cin>>t>>R;//t代表目标个数
R=min(5001,R);
n=m=5001;
while(t--){
int x,y,w;
cin>>x>>y>>w;
x++,y++;//++是为了从1开始计算,方便处理前缀和
s[x][y]+=w;//s[x][y]一开始是有值的,表示(x, y)这个点上的总权值,所以要用+=
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
s[i][j]+=s[i-1][j]+s[i][j-1]-s[i-1][j-1];//因为本题内存限制,可以把A数组省略,直接读入S中累加。
}
}
int res=0;
for(int i=R;i<=n;i++){
for(int j=R;j<=m;j++){
res=max(res,s[i][j]-s[i-R][j]-s[i][j-R]+s[i-R][j-R]);//枚举所有边长为R的正方形,枚举(i,j)为右下角
}
}
cout<<res<<endl;
return 0;
}
本题是要在一个矩形中(根据给出的x,y)放入多个正方形,枚举所有边长为R的正方形的右下角(i,y),这里的下标不是点的坐标,而是指格子的行号和列号,所以(i, j)表示的是第i行第j列的格子。s[i][j] - s[i - R][j] - s[i][j - R] + s[i - R][j - R]这里枚举的边长不是R,而是R - 1
看哪个正方形的价值最大。
该题涉及到容斥定理:
1.A[i,j]就等于位置(i,j)上所有目标W总,A[i,j]+=W
2.二维前缀和:s[i,j]=s[i-1][j]+s[i,j-1]-s[i-1][j-1]+A[i,j],即A[i,j]=s[i,j]-s[i-1][j]-s[i,j-1]+s[i-1][j-1]