GDUT寒假训练专题一 E激光炸弹

E - 激光炸弹

https://vjudge.csgrandeur.cn/problem/description/2126292156?1676609096000

思路：二维前缀和

用f[i][j]计算（1,1）到（i,j）所围成矩形的总和，则有：

f[i][j]=f[i-1][j]+f[i][j-1]-f[i-1][j-1]+a[i][j]

然后枚举正方形右下角坐标计算即可

代码：

#include<iostream>
#include<algorithm>
using namespace std;
int n,x,y,v,r,ans=0;
int f[5011][5011];
int main()
{
    cin>>n>>r;
    int a=r,b=r;
    for(int i=1;i<=n;i++)
    {
        cin>>x>>y>>v;
        x++;y++;
        f[x][y]+=v;
        a=max(a,x);b=max(b,y);
    }
    for(int i=1;i<=a;i++)
    for(int j=1;j<=b;j++)
    f[i][j]+=f[i-1][j]+f[i][j-1]-f[i-1][j-1];
    for(int i=r;i<=a;i++)
    for(int j=r;j<=b;j++)
    ans=max(ans,f[i][j]-f[i-r][j]-f[i][j-r]+f[i-r][j-r]);
    cout<<ans;
    return 0;
}