[백준 삼성 SW 역량 테스트 기출 문제] 17144번 미세먼지 안녕!(Python/파이썬)

삼성 SW 역량 테스트 기출 문제 - 17144번 미세먼지 안녕!

17144번: 미세먼지 안녕!

문제

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

공기청정기는 항상 1번 열에 설치되어 있고, 크기는 두 행을 차지한다. 공기청정기가 설치되어 있지 않은 칸에는 미세먼지가 있고, (r, c)에 있는 미세먼지의 양은 Ar,c이다.

1초 동안 아래 적힌 일이 순서대로 일어난다.

1. 미세먼지가 확산된다. 확산은 미세먼지가 있는 모든 칸에서 동시에 일어난다.
   • (r, c)에 있는 미세먼지는 인접한 네 방향으로 확산된다.
   • 인접한 방향에 공기청정기가 있거나, 칸이 없으면 그 방향으로는 확산이 일어나지 않는다.
   • 확산되는 양은 Ar,c/5이고 소수점은 버린다.
   • (r, c)에 남은 미세먼지의 양은 Ar,c - (Ar,c/5)×(확산된 방향의 개수) 이다.
2. 공기청정기가 작동한다.
   • 공기청정기에서는 바람이 나온다.
   • 위쪽 공기청정기의 바람은 반시계방향으로 순환하고, 아래쪽 공기청정기의 바람은 시계방향으로 순환한다.
   • 바람이 불면 미세먼지가 바람의 방향대로 모두 한 칸씩 이동한다.
   • 공기청정기에서 부는 바람은 미세먼지가 없는 바람이고, 공기청정기로 들어간 미세먼지는 모두 정화된다.

다음은 확산의 예시이다.

공기청정기의 바람은 다음과 같은 방향으로 순환한다.

방의 정보가 주어졌을 때, 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) 공기청정기의 위치를 찾아서 up, down 변수에 넣어준다 (공기청정기 함수에서 사용)

2) spread() : 미세먼지 확산 함수

• 배열 h에 확산된 미세먼지 양을 계산하여 담아주고 마지막에 기존 house 배열 원소에 확산 후 미세먼지 양을 반영해준다.

3) clean_up() : 공기청정기 윗부분

• dx, dy 배열을 반시계 방향으로 이동할 수 있게 선언
• 공기청정기 옆칸부터 시작하여 공기청정기 위치까지 while문이 반복된다.
• 범위 벗어나면 방향을 바꿔준다
• 바람 불면서 한 칸씩 이동해준다. 현재 위치 값과 이전에 저장했던 before 값을 서로 바꿔준다

4) clean_down() : 공기청정기 아랫부분

• dx, dy 배열을 시계 방향으로 이동할 수 있게 선언
• 공기청정기 옆칸부터 시작하여 공기청정기 위치까지 while문이 반복된다.
• 범위 벗어나면 방향을 바꿔준다
• 바람 불면서 한 칸씩 이동해준다. 현재 위치 값과 이전에 저장했던 before 값을 서로 바꿔준다

5) t초간 위의 과정을 진행한다

6) t초 후에 남아있는 미세먼지의 양을 모두 더하여 answer에 담고 출력한다

 

# 삼성 SW 역량 테스트 기출 문제
# 백준 17144번. 미세먼지 안녕!
# 구현, 시뮬레이션

r, c, t = map(int, input().split())
house = [list(map(int, input().split())) for i in range(r)]

up = -1
down = -1

# 공기청정기 위치 찾기
for i in range(r):
    # 0인 이유는 공기청정기는 항상 1번열에 설치되어있기 때문
    if house[i][0] == -1:
        up = i
        down = i + 1
        break

# 미세먼지 확산
def spread():
    dx = [0, -1, 0, 1]
    dy = [-1, 0, 1, 0]

    # 확산된 미세먼지 양을 담을 배열
    h = [[0] * c for _ in range(r)]

    for x in range(r):
        for y in range(c):
            if house[x][y] != 0 and house[x][y] != -1:
                cnt = 0     # 확산된 방향의 개수
                for i in range(4):
                    nx = x + dx[i]
                    ny = y + dy[i]

                    # 범위를 벗어나는 경우 혹은 공기청정기가 있는 경우엔 확산이 일어나지 않음
                    if nx < 0 or nx >= r or ny < 0 or ny >= c or house[nx][ny] == -1:
                        continue
                    else:
                        h[nx][ny] += house[x][y] // 5
                        cnt += 1
                h[x][y] += (house[x][y] - (house[x][y] // 5) * cnt)

    # 확산 후 미세먼지 양을 반영해준다
    for i in range(r):
        for j in range(c):
            if house[i][j] != -1:
                house[i][j] = h[i][j]


# 공기청정기 윗부분
def clean_up():
    dx = [0, -1, 0, 1]
    dy = [1, 0, -1, 0]

    i = 0
    before = 0
    x, y = up, 1    # 공기청정기 옆 부터 시작하므로 y는 1로 설정함

    while True:
        nx = x + dx[i]
        ny = y + dy[i]

        if x == up and y == 0:  # 공기청정기 위치와 같으면
            break
        if nx < 0 or nx >= r or ny < 0 or ny >= c:  # 범위 벗어나면 방향 바꿔줌
            i += 1
            continue

        # 바람불면 방향대로 한칸씩 이동
        house[x][y], before = before, house[x][y]
        x, y = nx, ny


def clean_down():
    dx = [0, 1, 0, -1]
    dy = [1, 0, -1, 0]

    i = 0
    before = 0
    x, y = down, 1

    while True:
        nx = x + dx[i]
        ny = y + dy[i]

        if x == down and y == 0:
            break
        if nx < 0 or nx >= r or ny < 0 or ny >= c:
            i += 1
            continue
        house[x][y], before = before, house[x][y]
        x, y = nx, ny



#t초 후 결과
for i in range(t):
    # 미세먼지 확산
    spread()

    # 공기청정기 작동
    clean_up()
    clean_down()

# t초 후 남아있는 미세먼지의 양 총합 출력
# 공기청정기 값 빼줘야하므로 2로 설정
answer = 2
for i in range(r):
    for j in range(c):
        answer += house[i][j]

print(answer)

알고리즘 분류

- 구현

- 시뮬레이션