[백준] 14499_주사위 굴리기 (Python)

📕 문제

https://www.acmicpc.net/problem/14499

14499번: 주사위 굴리기

 

💡 풀이법

1. dice의 각 6개 방향의 번호를 길이가 6인 리스트로 표현한다. (나는 0번 인덱스의 번호가 지도와 닿아있는 방향)

2. 동, 서, 남, 북으로 돌릴 때, 각 방향의 번호들을 바꿔준다.

3. 주사위를 놓는 곳이 0이라면 0번 인덱스의 숫자를 지도의 닿아있는 부분에 넣고, 반대로 0번 인덱스가 0이라면 지도의 숫자를 주사위에 넣는다.

 

N, M, x, y, K = map(int, input().split())
board = [list(map(int, input().split())) for _ in range(N)]
move = list(map(int, input().split()))
dxy = [(0, 1), (0, -1), (-1, 0), (1, 0)]
dice = [0, 0, 0, 0, 0, 0]

for m in move:
    dx, dy = dxy[m-1]
    x += dx
    y += dy
    if not 0 <= x < N or not 0 <= y < M:
        x -= dx
        y -= dy
        continue
    if m == 1:
        temp, dice[0] = dice[0], dice[2]
        temp, dice[3] = dice[3], temp
        temp, dice[5] = dice[5], temp
        dice[2] = temp
    elif m == 2:
        temp, dice[2] = dice[2], dice[0]
        temp, dice[5] = dice[5], temp
        temp, dice[3] = dice[3], temp
        dice[0] = temp
    elif m == 3:
        temp, dice[4] = dice[4], dice[0]
        temp, dice[5] = dice[5], temp
        temp, dice[1] = dice[1], temp
        dice[0] = temp
    else:
        temp, dice[0] = dice[0], dice[4]
        temp, dice[1] = dice[1], temp
        temp, dice[5] = dice[5], temp
        dice[4] = temp

    if not board[x][y]:
        board[x][y] = dice[0]
    else:
        dice[0] = board[x][y]
        board[x][y] = 0
        
    print(dice[5])