202104-2 邻域均值

C++:（前缀和）

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long LL;
const int N = 1010;
int a[N][N], cnt;
LL s[N][N];
int main()
{
    int n, L, r, t;
    int lf, ri, lo, hi;
    scanf("%d%d%d%d", &n, &L, &r, &t);
    for(int i = 1; i <= n; i++){
        for(int j = 1; j <= n; j++){
            scanf("%d", &a[i][j]);
            s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
            
        }
    }
    double avg;
    LL sum;
    int res = 0;
    for(int i = 1; i <= n; i++){
        for(int j = 1; j <= n; j++){
            if(j - r < 1) lf = 1;
            else lf = j - r;
            if(j + r > n) ri = n;
            else ri = j + r;
            if(i - r < 1) lo = 1;
            else lo = i  - r;
            if(i + r > n) hi = n;
            else hi = i + r;
            cnt = (ri - lf + 1) * (hi - lo + 1);
            sum = s[hi][ri] - s[lo - 1][ri] - s[hi][lf - 1] + s[lo - 1][lf - 1];
            avg = (double)sum / cnt;
            if(avg <= t) res++;
        }
    }
    cout << res << endl;
}