[Python] 백준 14499번 - 주사위 굴리기

📖 문제

14499번: 주사위 굴리기

 

🔎 풀이

백준 삼성 SW 역량 테스트 기출 문제집에 있는 문제로, 전형적인 구현 문제입니다. 가장 먼저 찾아야할 것은 주사위가 동서남북 각각으로 이동했을 때, 기존 위치와 어떻게 달라지는 가입니다. 아래 표는 문제에서 주어진 주사위 전개도를 각 방향으로 이동했을 때의 변화를 나타낸 표입니다.

방향 / 인덱스	1	2	3	4	5	6
1 (동쪽)	3	2	6	1	5	4
2 (서쪽)	4	2	1	6	5	3
3 (북쪽)	2	6	3	4	1	5
4 (남쪽)	5	1	3	4	6	2

 

 

빠른 시간 내에 해결하고자 한다면, 경우의 수가 많지 않기 때문에 아래 코드와 같이 방향에 따라 주사위 위치를 바꾸어주면 됩니다. 파이썬은 변수의 다중 할당을 편리하게 할 수 있습니다. 주사위가 이동하는 규칙만 설정한다면, 나머지 구현이 어려운 문제는 아니었습니다.

 

💻 코드

import sys
input=sys.stdin.readline

n, m, x, y, k = map(int, input().split())
grid = []
for _ in range(n):
    grid.append(list(map(int, input().split())))
orders = list(map(int, input().split()))

dice = [0] * 7
dx = [0, 0, -1, 1]
dy = [1, -1, 0, 0]

for i in range(k):
    order = orders[i]
    # 새로운 좌표 설정
    new_x = x + dx[order-1]
    new_y = y + dy[order-1]

    # 바깥으로 이동하는 경우 명령 무시
    if not (0 <= new_x < n) or not (0 <= new_y < m):
        continue
    
    # 주사위 이동
    if order == 1:
        dice[1], dice[3], dice[4], dice[6] = dice[3], dice[6], dice[1], dice[4]
    elif order == 2:
        dice[1], dice[3], dice[4], dice[6] = dice[4], dice[1], dice[6], dice[3]
    elif order == 3:
        dice[1], dice[2], dice[5], dice[6] = dice[2], dice[6], dice[1], dice[5]
    elif order == 4:
        dice[1], dice[2], dice[5], dice[6] = dice[5], dice[1], dice[6], dice[2]
    
    # 주사위의 바닥면과 칸에 쓰여 있는 수 변경
    if grid[new_x][new_y] == 0:
        grid[new_x][new_y] = dice[1]
    else:
        dice[1] = grid[new_x][new_y]
        grid[new_x][new_y] = 0
    
    # 좌표 저장
    x = new_x
    y = new_y

    print(dice[6])