o(n²)时间复杂度,暴力求解法,注意只能创建一个二维数组,创建两个就会爆内存,本题可以用扫描线写。
#include<iostream>
#include<cstring>
#include<cstdlib>
using namespace std;
const int N = 5000;
int n,r,map[N+5][N+5];
//int map[N+5][N+5];
int main()
{
ios::sync_with_stdio(false);
//memset(map,0,sizeof map);
//memset(map,0,sizeof map);
int x,y,v;
cin>>n>>r;
for(int i=1;i<=n;i++){
cin>>x>>y>>v;
map[x][y] = v;
}
for(int i=1;i<=N;i++) map[0][i] = map[0][i-1]+map[0][i];
for(int i=1;i<=N;i++) map[i][0] = map[i-1][0]+map[i][0];
for(int i=1;i<=N;i++){
for(int j=1;j<=N;j++){
map[i][j] = map[i-1][j]+map[i][j-1]-map[i-1][j-1]+map[i][j];
}
}
int maxn=0;
maxn = max(maxn,map[r-1][r-1]);
for(int i=r;i<=N;i++) maxn = max(maxn,map[r-1][i]-map[r-1][i-r]);
for(int i=r;i<=N;i++) maxn = max(maxn,map[i][r-1]-map[i-r][r-1]);
for(int i=r;i<=N;i++){
for(int j=r;j<=N;j++){
int temp = map[i][j]-map[i-r][j]-map[i][j-r]+map[i-r][j-r];
maxn = max(temp,maxn);
}
}
cout<<maxn;
return 0;
}