백준 온라인 저지, 구현 / 17144번: 미세먼지안녕 (파이썬 / 백준 골드문제, 삼성 SW 기출문제)

문제

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

 

문제 정의

미세먼지를 제거하기 위해 구사과는 공기청정기를 설치하려고 한다. 공기청정기의 성능을 테스트하기 위해 구사과는 집을 크기가 R×C인 격자판으로 나타냈고, 1×1 크기의 칸으로 나눴다. 구사과는 뛰어난 코딩 실력을 이용해 각 칸 (r, c)에 있는 미세먼지의 양을 실시간으로 모니터링하는 시스템을 개발했다. (r, c)는 r행 c열을 의미한다.

공기청정기는 항상 1번 열에 설치되어 있고, 크기는 두 행을 차지한다. 공기청정기가 설치되어 있지 않은 칸에는 미세먼지가 있고, (r, c)에 있는 미세먼지의 양은 Ar,c이다.
1초 동안 아래 적힌 일이 순서대로 일어난다.
다음은 확산의 예시이다.

왼쪽과 오른쪽에 칸이 없기 때문에, 두 방향으로만 확산이 일어났다.

인접한 네 방향으로 모두 확산이 일어난다.

공기청정기가 있는 칸으로는 확산이 일어나지 않는다.
공기청정기의 바람은 다음과 같은 방향으로 순환한다.

방의 정보가 주어졌을 때, T초가 지난 후 구사과의 방에 남아있는 미세먼지의 양을 구해보자.

입력

첫째 줄에 R, C, T (6 ≤ R, C ≤ 50, 1 ≤ T ≤ 1,000) 가 주어진다.
둘째 줄부터 R개의 줄에 Ar,c (-1 ≤ Ar,c ≤ 1,000)가 주어진다. 공기청정기가 설치된 곳은 Ar,c가 -1이고, 나머지 값은 미세먼지의 양이다. -1은 2번 위아래로 붙어져 있고, 가장 윗 행, 아랫 행과 두 칸이상 떨어져 있다.

출력

첫째 줄에 T초가 지난 후 구사과 방에 남아있는 미세먼지의 양을 출력한다.

예제 입력 1

7 8 1
0 0 0 0 0 0 0 9
0 0 0 0 3 0 0 8
-1 0 5 0 0 0 22 0
-1 8 0 0 0 0 0 0
0 0 0 0 0 10 43 0
0 0 5 0 15 0 0 0
0 0 40 0 0 0 20 0

예제 출력 1

188

접근 방법

- 1. 매 초마다 모든 미세먼지가 상하좌우로 확산되고 남은 미세먼지는 그 위치에 남는다.
- 1-1. 이를 확산시킬 때 모든 미세먼지가 퍼지고 난 뒤에 남은 미세먼지와 합쳐야지 제대로 된 확산이 이루어진다.
- 최대 연산 횟수: 칸의 개수 x 시간 x (상하좌우 연산 + 남은 미세먼지)= 50 x 50 x 1000 x 5 = 약 1250만번 이하(미세먼지 위치와 처음 주어지는 미세먼지의 양이 1000개이므로 이보다 적음)
- 2. 공기청정기가 순환하는 바람의 방향의 제일 끝부터 탐색을 진행하며 인덱스의 값을 한칸씩 이동시킨다.
- 최대 연산 횟수: 가로 세로 최대 길이 x 6 x 시간 = 50 x 6 x 1000 = 300,000회

코드

# https://www.acmicpc.net/problem/17144
# 접근 방법
# 1. 매 초마다 모든 미세먼지가 상하좌우로 확산되고 남은 미세먼지는 그 위치에 남는다.
# 1-1. 이를 확산시킬 때 모든 미세먼지가 퍼지고 난 뒤에 남은 미세먼지와 합쳐야지 제대로 된 확산이 이루어진다.
# 최대 연산 횟수: 칸의 개수 x 시간 x (상하좌우 연산 + 남은 미세먼지)= 50 x 50 x 1000 x 5 = 약 1250만번 이하(미세먼지 위치와 처음 주어지는 미세먼지의 양이 1000개이므로 이보다 적음)
# 2. 공기청정기가 순환하는 바람의 방향의 제일 끝부터 탐색을 진행하며 인덱스의 값을 한칸씩 이동시킨다.
# 최대 연산 횟수: 가로 세로 최대 길이 x 6 x 시간 = 50 x 6 x 1000 = 300,000회

import sys
r, c, t = map(int, sys.stdin.readline().split())
room = [list(map(int, sys.stdin.readline().split())) for _ in range(r)]
total_dust = 0
top_wind = [] # 윗 바람 방향의 역순 위치 저장
bottom_wind = [] # 아랫 바람 방향의 역순 위치 저장
for row in range(r):
    for col in range(c):
        if room[row][col] > 0:
            total_dust += room[row][col]
        elif room[row][col] == -1:
            i = row
            j = col
            if not top_wind:
                while i > 0:
                    i -= 1
                    top_wind.append([i, j])
                while j