BOJ 1895번 - 필터

[백준 온라인 저지 링크]

1895번: 필터

2차원 배열을 순회하며 3 * 3 크기의 부분배열을 구한 뒤, 해당 부분배열의 중간값들로 이루어진 배열을 생성합니다.

그리고 중간값들의 배열을 순회하면서 T보다 크거나 같은 값의 개수만 세주면 됩니다!

 

[정답 코드 - Python]

R, C = map(int, input().split())

image = []
filtered = []
cnt = 0


def findMid(R, C):
    global image
    pixels = []
    for row in range(3):
        for col in range(3):
            pixels.append(image[row + R][col + C])
    pixels.sort()
    return pixels[4]


for row in range(R):
    image.append(list(map(int, input().split())))

for row in range(R - 2):
    for col in range(C - 2):
        filtered.append(findMid(row, col))

T = int(input())
for i in filtered:
    if i >= T:
        cnt += 1

print(cnt)