原题链接:登录—专业IT笔试面试备考平台_牛客网
目录
1. 题目描述
2. 思路分析
3. 代码实现
1. 题目描述
2. 思路分析
二维前缀和板题。
注意从(1,1)开始存即可,所以每次输入x,y之后,要x++,y++。
因为m的范围最大为5000,每次又是一个边长为r的正方形,所以先用两重for循环求二维前缀和,再用两重循环(外层循环枚举矩阵右下角横坐标,内层循环枚举矩阵右下角纵坐标)求每个矩阵的和,然后不断更新最大值ans即可。
3. 代码实现
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
const int N=5010;
int a[N][N],pre[N][N];
signed main(){
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int n,r; cin>>n>>r;
while(n--){
int x,y,v; cin>>x>>y>>v;
x++,y++;
a[x][y]+=v;
}
for(int i=1;i<=5001;i++){
for(int j=1;j<=5001;j++){
pre[i][j]=a[i][j]+pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1];
}
}
int ans=0;
for(int i=r;i<=5001;i++){
for(int j=r;j<=5001;j++){
int x1=i-r+1,y1=j-r+1;
int x2=i,y2=j;
int tmp=pre[x2][y2]-pre[x2][y1-1]-pre[x1-1][y2]+pre[x1-1][y1-1];
ans=max(ans,tmp);
}
}
cout<<ans<<endl;
return 0;
}