백준 알고리즘 17144번 미세먼지 안녕! 풀이 With Python

문제 링크: https://www.acmicpc.net/problem/17144

17144번: 미세먼지 안녕!

미세먼지 안녕! 문제는 크게 2가지 문제를 해결해야 합니다. 첫 번째 문제는 미세먼지가 확산하는 것을 구현해야 하는데, BFS(너비 우선 탐색)을 사용하여 해결할 수 있습니다. 두 번째 문제는 공기청정기가 미세먼지를 미는 것을 구현해야 하는데, 이는 for문 혹은 while문을 사용하여 해결할 수 있습니다. 소스코드는 아래와 같습니다. 

 

import sys


dx = [-1, 0, 1, 0]
dy = [0, -1, 0, 1]
time = 0
map1 = []
cleaner = []
cleandic = {}
r, c, t = map(int, sys.stdin.readline().split())
for i in range(r):
    s = list(map(int, sys.stdin.readline().split()))
    for j in range(c):
        if s[j] == -1:
            cleaner.append((i, j))
            cleandic[(i, j)] = True
    map1.append(s)

while time < t:
    map2 = [[0] * c for _ in range(r)]
    for i in range(r):
        for j in range(c):
            if map1[i][j] > 0:
                count = 0
                for d in range(4):
                    nx = i + dx[d]
                    ny = j + dy[d]
                    if -1 < nx < r and -1 < ny < c and cleandic.get((nx, ny)) is None:
                        count += 1
                        map2[nx][ny] += map1[i][j] // 5
                map2[i][j] += map1[i][j] - count * (map1[i][j] // 5)
    for i in range(2):
        temp = 0
        temp2 = 0
        if i == 0:
            dx1 = [0, -1, 0, 1]
            dy1 = [1, 0, -1, 0]
            x = cleaner[0][0]
            y = cleaner[0][1] + 1
            d = 0
            while x != cleaner[0][0] or y != cleaner[0][1]:
                temp2 = map2[x][y]
                map2[x][y] = temp
                temp = temp2
                if 0 > x + dx1[d] or x + dx1[d] > r - 1 or y + dy1[d] < 0 or y + dy1[d] > c - 1:
                    d += 1
                x += dx1[d]
                y += dy1[d]
        else:
            dx2 = [0, 1, 0, -1]
            dy2 = [1, 0, -1, 0]
            x = cleaner[1][0]
            y = cleaner[1][1] + 1
            d = 0
            while x != cleaner[1][0] or y != cleaner[1][1]:
                temp2 = map2[x][y]
                map2[x][y] = temp
                temp = temp2
                if 0 > x + dx2[d] or x + dx2[d] > r - 1 or y + dy2[d] < 0 or y + dy2[d] > c - 1:
                    d += 1
                x += dx2[d]
                y += dy2[d]
    map1=map2
    time += 1
answer = 0
for i in range(r):
    for j in range(c):
        if map1[i][j] > 0:
            answer += map1[i][j]
print(answer)

소스코드에 대한 질문이 있으시다면, 댓글을 남겨주시면 답변을 드리겠습니다. 읽어주셔서 감사합니다. :-)