[BOJ] 14499 - 주사위 굴리기 (Python)

 

 

아이디어

 

조건에 따라 주사위가 이동할 때마다 주사위와 지도의 상태를 변경한다.

 

 

풀이

 

import sys

input = sys.stdin.readline

delta = [(0, 1), (0, -1), (-1, 0), (1, 0)]
dice = [0]*6
turned = [
    (3, 1, 0, 5, 4, 2),
    (2, 1, 5, 0, 4, 3),
    (4, 0, 2, 3, 5, 1),
    (1, 5, 2, 3, 0, 4)
]

def solution():
    N, M, x, y, K = map(int, input().split())
    board = [list(map(int, input().split())) for _ in range(N)]
    for d in map(int, input().split()):
        d -= 1
        nx, ny = x+delta[d][0], y+delta[d][1]
        if N > nx >= 0 and M > ny >= 0:
            temp = dice[:]
            for i in range(6):
                dice[i] = temp[turned[d][i]]
            x, y = nx, ny
            if board[x][y]:
                dice[5] = board[x][y]
                board[x][y] = 0
            else:
                board[x][y] = dice[5]
            print(dice[0])

solution()