HYSBZ -1218 激光炸弹

#include <iostream>
#include <algorithm>
#include <cmath>
#include <stdio.h>
#include <cstring>
#include <string>
#include <cstdlib>
#include <queue>
#include <stack>
#include <set>
#include <vector>
#include <map>
#include <list>
#include <iomanip>
 #include <fstream>
using namespace std;

int a[5003][5003];

int main()
{
    int n,r,x,y,v,nn,mm; 
    
    scanf("%d%d",&n,&r);
    nn=mm=r;
    //读入目标价值 
    for(int i=0;i<n;++i)
    {
        scanf("%d%d%d",&x,&y,&v);
        ++x;
        ++y;
        a[x][y]+=v;
        nn=max(nn,x);
        mm=max(mm,y);
    }
    //计算前缀和 
    for(int i=1;i<=nn;++i)
    {
        for(int j=1;j<=mm;++j)
        {
            a[i][j]+=a[i][j-1]+a[i-1][j]-a[i-1][j-1];
        }
    }
    //计算最大前缀和，但是要小于边长r 
    int ans=0,t;
     for(int i=r;i<=nn;++i)
     {
         for(int j=r;j<=mm;++j)
         {
             t=a[i][j]-a[i-r][j]-a[i][j-r]+a[i-r][j-r];
             ans=max(ans,t);
         }
     }
     printf("%d
",ans);
     
    return 0;
}