[BOJ_14499/Python] 주사위 굴리기

문제

14499번: 주사위 굴리기

14499번: 주사위 굴리기

 

---

풀이

import sys
input = sys.stdin.readline


def roll_the_dice(m):
    if m == 1:  # 동
        return [dice[0], dice[2], dice[3], dice[5], dice[4], dice[1]]
    if m == 2:  # 서
        return [dice[0], dice[5], dice[1], dice[2], dice[4], dice[3]]
    if m == 3:  # 남
        return [dice[2], dice[1], dice[4], dice[3], dice[5], dice[0]]
    if m == 4:  # 북
        return [dice[5], dice[1], dice[0], dice[3], dice[2], dice[4]]


# 입력
N, M, x, y, K = map(int, input().split())
arr = [list(map(int, input().split())) for _ in range(N)]
move = list(map(int, input().split()))

# 풀이
dice = [0, 0, 0, 0, 0, 0]
dir = [(0, 0), (0, 1), (0, -1), (-1, 0), (1, 0)]

btm_idx, top_idx = 2, 5
for m in move:
    dx, dy = dir[m]

    if 0 <= x + dx <= N - 1 and 0 <= y + dy <= M - 1:
        x = x + dx
        y = y + dy

        dice = roll_the_dice(m)

        if arr[x][y] != 0:  # 1. 칸에 쓰인 수 주사위 바닥면으로 복사, 칸에 쓰인 수 = 0
            dice[btm_idx] = arr[x][y]
            arr[x][y] = 0
        else:  # 2. 주사위 바닥면의 수 칸에 복사
            arr[x][y] = dice[btm_idx]

        # 출력
        print(dice[top_idx])

roll_the_dice() 함수를 만들어서 주사위를 굴리는 동작을 처리하게 했다. 문제에서 제시한 도면을 기준으로 다음과 같이 인덱스를 잡았다.

 

주사위를 동서남북으로 굴리더라도 2번 인덱스는 무조건 바닥면에 쓰인 값을, 5번 인덱스는 무조건 윗 면에 쓰인 값을 나타내도록 고정해놨기 때문에 어디가 바닥면이고 어디가 윗 면인지 고민할 필요가 없어서 좋다.