邻域均值 （python）

试题背景

顿顿在学习了数字图像处理后，想要对手上的一副灰度图像进行降噪处理。不过该图像仅在较暗区域有很多噪点，如果贸然对全图进行降噪，会在抹去噪点的同时也模糊了原有图像。因此顿顿打算先使用邻域均值来判断一个像素是否处于较暗区域，然后仅对处于较暗区域的像素进行降噪处理。

问题描述

计算机软件能力认证考试系统http://118.190.20.162/view.page?gpid=T127

输入格式

输入共 n+1 行。

输入的第一行包含四个用空格分隔的正整数 n、L、r 和 t，含义如前文所述。

第二到第 n+1 行输入矩阵 A。
第 i+2（0≤i<n）行包含用空格分隔的 n 个整数，依次为 Ai0,Ai1,⋯,Ai(n−1)。

输出格式

输出一个整数，表示输入灰度图像中处于较暗区域的像素总数。

样例输入

4 16 1 6
0 1 2 3
4 5 6 7
8 9 10 11
12 13 14 15

Data

样例输出

7

Data

样例输入

11 8 2 2
0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0 0
0 7 0 0 0 7 0 0 7 7 0
7 0 7 0 7 0 7 0 7 0 7
7 0 0 0 7 0 0 0 7 0 7
7 0 0 0 0 7 0 0 7 7 0
7 0 0 0 0 0 7 0 7 0 0
7 0 7 0 7 0 7 0 7 0 0
0 7 0 0 0 7 0 0 7 0 0
0 0 0 0 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0 0 0 0

Data

样例输出

83

Data

评测用例规模与约定

70% 的测试数据满足 n≤100、r≤10。

全部的测试数据满足 0<n≤600、0<r≤100 且 2≤t<L≤256。

代码

#参考二维前缀和 https://blog.csdn.net/justidle/article/details/103754960
n, L, r, t = map(int,input().split())
nums = []
sum = [[0]*(n+1) for _ in range(n+1)]
for i in range(n):
    nums.append(list(input().split(' ')))

for i in range(1,n+1):
    for j in range(1,n+1):
        sum[i][j] = sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1] + int(nums[i-1][j-1])

# for i in range(1,n+1):
#     for j in range(1,n+1):
#         print(sum[i][j], end=' ')
#     print()

# x-r,y-r -> x,y -> x+r,y+r
count = 0
for i in range(1,n+1):
    for j in range(1,n+1):
        x1,x2 = i-r, i+r
        y1,y2 = j-r, j+r
        #判断是否越界
        if x1 < 1: x1 = 1
        if x2 > n: x2 = n
        if y1 < 1: y1 = 1
        if y2 > n: y2 = n
        # print(x1,y1,x2,y2,'-----', sum[x2][y2] - sum[x1-1][y2] - sum[x2][y1-1] + sum[x1-1][y1-1], t*(x2-x1+1)*(y2-y1+1))
        if sum[x2][y2] - sum[x1-1][y2] - sum[x2][y1-1] + sum[x1-1][y1-1] <= t*(x2-x1+1)*(y2-y1+1):
            count += 1
print(count)