ccf 202104-2 邻域均值

第二题   

1. 利用vector<vector<int>> mp(n,vector<int>(n));这个数据结构来存储前缀和

2. 可以再次优化，将A去掉

//
// Created by clay on 2021/4/23.
//

#include <bits/stdc++.h>
using namespace std;

int main(){
    int n,L,r,t;
    cin>>n>>L>>r>>t;
    //定义区间和数据结构: 只需要知道区间和即可
    vector<vector<int>> mp(n,vector<int>(n));
    vector<vector<int>> A(n,vector<int>(n));
    for(int i=0;i<A.size();i++)
        for(int j=0;j<A[0].size();j++){
            cin>>A[i][j];
            if(j==0) mp[i][j]=A[i][j];
            else
                mp[i][j]=mp[i][j-1]+A[i][j];
        }
    int sum=0;
    for(int i=0;i<A.size();i++)
        for(int j=0;j<A[0].size();j++){
            //越界元素的处理
            int min,min2,max,max2;
            min=i-r<0?0:i-r;
            min2=j-r<0?0:j-r;
            max=i+r<=A.size()-1?i+r:A.size()-1;
            max2=j+r<=A[0].size()-1?j+r:A[0].size()-1;
            int s=(max-min+1)*(max2-min2+1);
            int constrain=t*s;
            double d=0; //存储最终的结果
            for(int k=min;k<=max;k++){
                if(min2==0)
                    d+=mp[k][max2];
                else
                    d+=mp[k][max2]-mp[k][min2-1];
                if(d>constrain)
                    break;
            }
            if(d<=constrain)
                sum++;
        }
    cout<<sum;
    return 0;
}