-
[백준 17144] 미세먼지 안녕!Algorithm/Source Code 2022. 1. 29. 13:16반응형
문제
미세먼지를 제거하기 위해 구사과는 공기청정기를 설치하려고 한다. 공기청정기의 성능을 테스트하기 위해 구사과는 집을 크기가 R×C인 격자판으로 나타냈고, 1×1 크기의 칸으로 나눴다. 구사과는 뛰어난 코딩 실력을 이용해 각 칸 (r, c)에 있는 미세먼지의 양을 실시간으로 모니터링하는 시스템을 개발했다. (r, c)는 r행 c열을 의미한다.
공기청정기는 항상 1번 열에 설치되어 있고, 크기는 두 행을 차지한다. 공기청정기가 설치되어 있지 않은 칸에는 미세먼지가 있고, (r, c)에 있는 미세먼지의 양은 Ar,c이다.
1초 동안 아래 적힌 일이 순서대로 일어난다.
- 미세먼지가 확산된다. 확산은 미세먼지가 있는 모든 칸에서 동시에 일어난다.
- (r, c)에 있는 미세먼지는 인접한 네 방향으로 확산된다.
- 인접한 방향에 공기청정기가 있거나, 칸이 없으면 그 방향으로는 확산이 일어나지 않는다.
- 확산되는 양은 Ar,c/5이고 소수점은 버린다.
- (r, c)에 남은 미세먼지의 양은 Ar,c - (Ar,c/5)×(확산된 방향의 개수) 이다.
- 공기청정기가 작동한다.
- 공기청정기에서는 바람이 나온다.
- 위쪽 공기청정기의 바람은 반시계방향으로 순환하고, 아래쪽 공기청정기의 바람은 시계방향으로 순환한다.
- 바람이 불면 미세먼지가 바람의 방향대로 모두 한 칸씩 이동한다.
- 공기청정기에서 부는 바람은 미세먼지가 없는 바람이고, 공기청정기로 들어간 미세먼지는 모두 정화된다.
다음은 확산의 예시이다.
왼쪽과 오른쪽에 칸이 없기 때문에, 두 방향으로만 확산이 일어났다.
인접한 네 방향으로 모두 확산이 일어난다.
공기청정기가 있는 칸으로는 확산이 일어나지 않는다.
공기청정기의 바람은 다음과 같은 방향으로 순환한다.
방의 정보가 주어졌을 때, 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초가 지난 후 구사과 방에 남아있는 미세먼지의 양을 출력한다.
solution.py
r, c, t = map(int, input().split()) room = [list(map(int, input().split())) for _ in range(r)] # 공기청정기 위치 cleaner = [] for i in range(r): for j in range(c): if room[i][j] == -1: cleaner.append((i, j)) # 미세먼지 확산 def diffusion(): global room, cleaner temp = [[0]*c for _ in range(r)] move = [(0,1),(0,-1),(1,0),(-1,0)] for i in range(r): for j in range(c): if room[i][j]==0 or room[i][j]==-1: continue cnt = 0 for x, y in move: cx = i+x cy = j+y if 0<=cx<r and 0<=cy<c and room[cx][cy]!= -1: temp[cx][cy] += room[i][j]//5 cnt += 1 temp[i][j] += room[i][j]-cnt*(room[i][j]//5) temp[cleaner[0][0]][cleaner[0][1]] = -1 temp[cleaner[1][0]][cleaner[1][1]] = -1 room = temp # 공기청정기 작동 def clean(x, y, move): last = 0 j = 0 start = (x, y) x += move[0][0] y += move[0][1] while x != start[0] or y != start[1]: next = room[x][y] room[x][y] = last last = next x += move[j][0] y += move[j][1] if not (0<=x<r and 0<=y<c): if j == 3: break x = x - move[j][0] + move[j+1][0] y = y - move[j][1] + move[j+1][1] j += 1 for _ in range(t): diffusion() clean(cleaner[0][0], cleaner[0][1], [(0, 1), (-1, 0), (0, -1), (1, 0)]) clean(cleaner[1][0], cleaner[1][1], [(0, 1), (1, 0), (0, -1), (-1, 0)]) answer = 0 for i in range(r): for j in range(c): if room[i][j] and room[i][j] != -1: answer+= room[i][j] print(answer)
출처
https://www.acmicpc.net/problem/17144
반응형'Algorithm > Source Code' 카테고리의 다른 글
[백준 11559] Puyo Puyo (0) 2022.02.05 [백준 2573] 빙산 (0) 2022.02.04 [백준 16234] 인구 이동 (0) 2022.01.28 [백준 2230] 수 고르기 (0) 2022.01.28 [프로그래머스 42842] (0) 2022.01.27 - 미세먼지가 확산된다. 확산은 미세먼지가 있는 모든 칸에서 동시에 일어난다.