[백준/Python] #1895 필터

문제

 

https://www.acmicpc.net/problem/1895

1895번: 필터

---

Algorithm

2차원 배열 A와 3x3 크기의 필터 K에 대해 B의 좌측 상단 부분을 A의 좌측 상단에 맞추고 필터 K를 오른쪽으로 한 칸씩 옮기고 오른쪽으로 더이상 옮길 수 없을 때는 아래로 한 칸 옮기고 아래로도 움직일 수 없을 때까지 같은 과정을 반복한다. 이 과정을 반복하는 동안 K와 A가 겹치는 영역에 해당하는 원소들로 이루어진 list를 만들고 K가 움직이면서 만든 list들의 중앙값들 중 기준값 T보다 크거나 같은 것들의 개수를 출력한다.

 

---

 

 

Code

import sys

input = sys.stdin.readline

R, C = map(int, input().split())
array = []
for _ in range(R):
    array.append(list(map(int, input().split())))
T = int(input())

ans = 0
for i in range(R - 2):
    rows = array[i : i + 3]
    for j in range(C - 2):
        box = []
        box += rows[0][j : j + 3]
        box += rows[1][j : j + 3]
        box += rows[2][j : j + 3]
        box.sort()
        if box[4] >= T:
            ans += 1

print(ans)