哇,一维都不会突然就二维了哇
但其实思想就是对前面的数进行一个O(n)统计
而二维的规律看图:
可以发现 sum[i,j] = sum[i,j-1] + sum[i-1,j] - sum[i-1,j-1] +a[i,j] (a[i,j] 为那个位置的数)
然后题目炸一个R的正方形,那不就是举一反三嘛,来看图
(红色的正方形)= sum[i,j] - sum[i,j-r] - sum[i-r,j] + sum[i-r,j-r]
不就是有手就行?!
哦题目有两个坑:
- 位置可能重复,而你需要叠加该位置的数!
- R可能比n,m大,而且大到了10^9,所以我们在最后的时候,就判断一下行是R大还是n大,列是R大还是m大,谁大谁就让小的好吧,意会一下就懂啦
代码如下:
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 5050;//最大n,m
int sum[N][N];
int main()
{
int N,R;
scanf("%d%d",&N,&R);
int n = 0,m = 0;
for (int i = 0,x,y,w;i < N; i++){
scanf("%d%d%d",&x,&y,&w);
x++; y++;
sum[x][y] += w;//叠加
n = max(x,n);
m = max(y,m);
//找最大n,m
}
//求前缀和
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
{
sum[i][j] += sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1];
}
int res = 0;
//求最牛逼的正方形
int ic = min(R,n),jc = min(R,m);
for (int i = ic; i <= n; i++){
for (int j = jc; j <= m; j++){
res = max (res,sum[i][j]-sum[i-ic][j]-sum[i][j-jc]+sum[i-ic][j-jc]);
}
}
printf("%d",res);
return 0;
}