洛谷 P2038 无线网络发射器选址

P2038 无线网络发射器选址

注意边界是否小于0

#include<bits/stdc++.h>
using namespace std;
int d,n,num[1000][1000],ans[1000][1000],tot,cnt,maxx;
int main()
{
    scanf("%d",&d);
    scanf("%d",&n);
    memset(num,0,sizeof(num));
    memset(ans,0,sizeof(ans));
    for(int i=1;i<=n;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        num[x][y]=z;
    }
    for(int i=0;i<=128;i++)
        for(int j=0;j<=128;j++)
        {            
            if(i-d>=0&&j-d>=0)
            for(int k=i-d;k<=i+d;k++)
                for(int l=j-d;l<=j+d;l++)    
                    ans[i][j]+=num[k][l];
            else{
                if(i-d<0&&j-d>=0)
                    for(int k=0;k<=i+d;k++)
                        for(int l=j-d;l<=j+d;l++)
                            ans[i][j]+=num[k][l];
                if(i-d>=0&&j-d<0)
                    for(int k=i-d;k<=i+d;k++)
                        for(int l=0;l<=j+d;l++)
                            ans[i][j]+=num[k][l];
                if(i-d<0&&j-d<0)
                    for(int k=0;k<=i+d;k++)
                        for(int l=0;l<=j+d;l++)
                            ans[i][j]+=num[k][l];
                
            }    
            maxx=max(ans[i][j],maxx);
        }
    for(int i=0;i<=128;i++)
        for(int j=0;j<=128;j++)
            if(ans[i][j]==maxx) tot++;
    printf("%d %d",tot,maxx);
    
}