首先能看出来是个二维前缀和
有几个细节:
1、r如果比5000大,要特判,而且要处理重复点
2、r如果比你给的所有坐标都要大,那么要处理一下,也就是要你row和col都大于r才能求解,所以一开始row和col的最小值要设置成r
3、题目中说了边上的点不算,yxc直播的讲解我觉得不太对,进阶指南的说法应该是正确的,就是正常求解一个矩形的权值和的方法和题目要求的边点不算其实是等价的,也就是说:
这个求解方法可以等价于去掉了边点,
如下图,求解以(2, 2)为右下角的边长为1的矩形覆盖,是map_[2][2]-map[2][1]-map_[1][2]+map_[1][1]
,这个相当于去掉了加粗的两条黑边,也相当于一个同样大小的矩形平移了一下之后的覆盖(虚线矩形),所以实际上是等价于题意的。yxc的讲解认为如果算边点,那么应该要+1,我想了半天觉得没道理。
AC代码:
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include <set>
#include <vector>
using namespace std;
int map_[5000+10][5000+10];
int n, r;
int col, row;
int main() {
cin >> n >> r;
col = row = r;
int res = 0;
for (int i = 0; i < n; i++) {
int x, y, w;
cin >> x >> y >> w;
map_[x+1][y+1] += w; //重复点
row = max(row, x+1);
col = max(col, y+1);
res+=w;
}
if(r>=5000) { //如果矩形比最大范围都大,直接算全部的和就行
cout << res << endl;
return 0;
}
for (int i = 1; i <= row; i++) {
for (int j = 1; j <= col; j ++) {
map_[i][j] += (map_[i-1][j] + map_[i][j-1] - map_[i-1][j-1]);
}
}
int ans = -1;
for (int i = r; i <= row; i ++) {
for (int j = r; j <= col; j++) {
ans = max(ans, map_[i][j] - map_[i-r][j] - map_[i][j-r] + map_[i-r][j-r]);
}
}
cout << ans << endl;
return 0;
}