백준: 14499 주사위 굴리기

문제

14499번: 주사위 굴리기

 

문제 풀이

 초기에 6면이 0의 값을 가지는 주사위를 동, 서, 북, 남으로 지도를 탐색할 경우 현재 좌표의 지도 값이 0이 아니면 주사위 바닥에 값을 복사하고, 현재 좌표 지도 값이 0인 경우 주사위 바닥의 수를 지도에 복사한다. 이때 명령에 따라 주사위를 움직일 때, 주사위 윗면의 값을 출력하는 문제이다.

 

 

 주사위는 동, 서, 북, 남 각 방향에 따라 그림과 같이 변화한다. 동, 서인 경우에는 기존의 4, 1, 3, 6번이 오른쪽으로 순회하거나 왼쪽으로 순회한다. 이와 달리 북, 남인 경우 2, 1, 5, 6번이 위나 아래로 순회하게 된다. 따라서 이와 같은 경우를 고려하여 주사위가 해당 방향으로 이동 가능한 경우 주사위 상태를 변화시키면서 탐색을 진행하면 문제를 해결할 수 있다.

 

코드

from sys import stdin
from collections import deque


def move_dice(direction):
    if direction == E:
        dice[3], dice[1], dice[4], dice[6] = dice[1], dice[4], dice[6], dice[3]
    elif direction == W:
        dice[3], dice[1], dice[4], dice[6] = dice[6], dice[3], dice[1], dice[4]
    elif direction == N:
        dice[2], dice[1], dice[5], dice[6] = dice[1], dice[5], dice[6], dice[2]
    elif direction == S:
        dice[2], dice[1], dice[5], dice[6] = dice[6], dice[2], dice[1], dice[5]


def bfs(start):
    q = deque([start])
    # 동, 서, 북, 남
    dirs = [(), (0, 1), (0, -1), (-1, 0), (1, 0)]

    for order in orders:
        cur_x, cur_y = q.popleft()
        dx, dy = dirs[order]
        next_x, next_y = cur_x + dx, cur_y + dy

        if 0 <= next_x < n and 0 <= next_y < m:
            move_dice(order)
            if graph[next_x][next_y]:
                dice[6] = graph[next_x][next_y]
                graph[next_x][next_y] = 0
            else:
                graph[next_x][next_y] = dice[6]
            print(dice[1])
            q.append((next_x, next_y))
        else:
            q.append((cur_x, cur_y))


if __name__ == '__main__':
    E, W, N, S = 1, 2, 3, 4
    n, m, x, y, k = map(int, stdin.readline().split())
    graph = [list(map(int, stdin.readline().split())) for _ in range(n)]
    orders = list(map(int, stdin.readline().split()))
    # 주사위의 인덱스는 문제에 주어진 것과 동일.
    dice = [0] * 7
    bfs([x, y])

 예제 입력 케이스만 모두 잘 처리 된다면, 답안을 제출하였을 때 정답이라는 문구를 볼 수 있다. 동, 서, 남, 북이 아니라 동, 서, 북, 남이라서 문제를 풀면서 한번 헷갈렸다. 또한, 주사위를 동, 서로 옮길 때 바닥면도 포함해서 순회된다는 것을 인지하지 못해 한번 더 헷갈렸다.😂