Point 💡

• 차례차례 값을 확인하고, 지도에 입력하는 작업을 하면, 원래 확산시키려는 먼지의 양이 변경된다.

  → 차례차례 확산되는 것이 아니라 한꺼번에 확산이 되어야하므로, 먼지의 양을 그대로 유지하기위해 먼지 리스트를 만들어서 함수에 넘겨준다.

• 공기청정기를 작동하는 작업은 앞에서부터하면 3, 4, 6 → 3, 3, 6 → 3, 3, 3 다음과 같이 값이 변동이 일어난다고 생각했다.

  → 뒤에서부터 작업하면, 3, 4, 6, 7, 8 → 3, 4, 6, 7, 7 → 3, 4, 6, 6, 7 → 3, 4, 4, 6, 7 ... 으로 정상적으로 작동된다.

  • 이때는 temp 변수 없이도 room[i][0] = room[i-1][0] 다음과 같이 코드를 작성해도 값의 변경이 일어나지 않는다.
  • 경계값은 각 경우마다 위, 아래, 좌, 우로 이동하도록 코드를 짜주었다.

  하나씩 범위를 고려하다보니까 시간이 오래 걸렸다. 다른 방법이 없을까?

  1. 공기청정기의 오른쪽에서 시작하고 각각의 방향에 따라서 index를 조정해준다.

  2. 만약 x,y로 이동하다가 벽에 닿는다면 index를 조정하여 방향을 바꿔준다.

     → 이 부분이 키 포인트이다!

  3. swap방식으로 temp에 값을 저장하고 이를 다음 값에 전달하는 형태로 한칸씩 미뤄주었다.

     → 어? 생각해보니 어짜피 temp를 사용한다면, 굳이 뒤에서부터 안해도되네?

     → temp를 사용하는 방식을 잘 기억해두자!

dx = [-1,0,1,0] # 상 우 하 좌
dy = [0,1,0,-1]  
def clean(a,b):
    # 위에 회전
    x, y = a, 1
    index = 1 # 우부터 시작 우 상 좌 하
    temp = 0 # 공기청정기에서 나오는 바람
    while True:
        nx = x+dx[index]
        ny = y+dy[index]
        if nx==R or ny == C or nx==-1 or ny==-1: # 벽에 닿았을 때
            index = (index-1)%4
            continue
        if x==a and y==0: # 공기청정기로 다시 돌아옴
            break
        pan[x][y], temp = temp, pan[x][y] # swap
        x,y = nx, ny
    # 아래 회전
    x, y = b, 1
    index = 1 # 우부터 시작 우 하 좌 상
    temp = 0 # 공기청정기에서 나오는 바람
    while True:
        nx = x+dx[index]
        ny = y+dy[index]
        if nx==R or ny == C or nx==-1 or ny==-1: # 벽에 닿았을 때
            index = (index+1)%4
            continue
        if x==b and y==0: # 공기청정기로 다시 돌아옴
            break
        pan[x][y], temp = temp, pan[x][y] # swap
        x,y = nx, ny

My Solution

# 확산
def spread(dust):
    for i in range(len(dust)):
        amount = dust[i][2] // 5
        cnt = 0
        for j in range(4):
            nx = dust[i][0] + dx[j]
            ny = dust[i][1] + dy[j]

            if nx < 0 or nx >= r or ny < 0 or ny >= c or room[nx][ny] == -1:
                continue

            room[nx][ny] += amount
            cnt += 1

        room[dust[i][0]][dust[i][1]] -= amount * cnt

# 공기청정기 작동 : 뒤에서부터?
def work(x, mode):

    if mode == 0:

        for i in range(x-1, 0, -1):
            room[i][0] = room[i-1][0]

        for i in range(c-1):
            room[0][i] = room[0][i+1]

        for i in range(x):
            room[i][c-1] = room[i+1][c-1]

        for i in range(c-1, 0, -1):
            if room[x][i-1] == -1:
                room[x][i] = 0
                break
            room[x][i] = room[x][i-1]

    else:

        for i in range(x+1, r-1):
            room[i][0] = room[i+1][0]

        for i in range(c-1):
            room[r-1][i] = room[r-1][i+1]

        for i in range(r-1, x, -1):
            room[i][c-1] = room[i-1][c-1]

        for i in range(c-1, 0, -1):
            if room[x][i-1] == -1:
                room[x][i] = 0
                break
            room[x][i] = room[x][i - 1]

import sys
input = sys.stdin.readline
r, c, t = map(int, input().split())
room = [list(map(int, input().split())) for _ in range(r)]

dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]

for i in range(t):
    dust = []
    for j in range(r):
        for k in range(c):
            if room[j][k] != 0 and room[j][k] != -1:
                dust.append((j, k, room[j][k]))

    spread(dust)

    index = 0
    for j in range(r):
        if room[j][0] == -1:
            work(j, index)
            index += 1

answer = 0
for i in range(r):
    answer += sum(room[i])

# -1 2개 빼주기
print(answer+2)