알고리즘/알고리즘문풀

BOJ - 주사위 굴리기 14499번 (python)

developer-ellen 2021. 10. 14. 11:35

❓ 문제 - 백준 주사위 굴리기 14499번 - python 풀이법

출처 

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

 

14499번: 주사위 굴리기

첫째 줄에 지도의 세로 크기 N, 가로 크기 M (1 ≤ N, M ≤ 20), 주사위를 놓은 곳의 좌표 x y(0 ≤ x ≤ N-1, 0 ≤ y ≤ M-1), 그리고 명령의 개수 K (1 ≤ K ≤ 1,000)가 주어진다. 둘째 줄부터 N개의 줄에 지도

www.acmicpc.net

 

 

📝 문제해결법

1. 이 문제는 구현으로 풀었다.

  • 주사위의 윗면-뒤쪽-오른쪽-왼쪽-앞쪽-바닥면에 대한 값들을 dice 인덱스를 통해 저장
  • 지도의 값을  board를 통해 저장한다.
  • 주사위를 움직일 때 dx, dy(동-0, 서-1, 북-2, 남-3)를 이용해서 움직이려는 곳이 지도 밖이면 움직이지 않게 한다.
  • 주사위를 움직일 때 동,서,북,남 방향에 맞춰서 move_east(), move_west(), move_north(), move_south() 함수를 호출하여 주사위를 해당 방향으로 움직인다.
  • 만약 움직였을 때 해당 지도의 값이 0이면 주사위의 바닥면(dice[-1])의 값을 지도에 복사한다.
  • 만약 움직였을 때 해당 지도의 값이 0이 아니면 바닥면(dice[-1])에 지도의 값을 복사한 후 지도 값을 0으로 변경한다.

2. 주사위 굴림 구현

  • move_east() : 주사위 동쪽으로 굴리면 주사위의 값들이 바닥면->왼쪽->윗면->오른쪽->바닥면 으로 이동하는 것을 구현한다.
  • move_west() : 주사위 서쪽으로 굴리면 주사위의 값들이 바닥면->오른쪽->윗면->왼쪽->바닥면 으로 이동하는 것을 구현한다.
  • move_north() : 주사위 북쪽으로 굴리면 주사위의 값들이 바닥면->앞쪽->윗면->뒤쪽->바닥면 으로 이동하는 것을 구현한다.
  • move_south() : 주사위 남쪽으로 굴리면 주사위의 값들이 바닥면->뒤쪽->윗면->앞쪽->바닥면 으로 이동하는 것을 구현한다.

 

 

💻 소스코드

# 주사위 굴리기 - BOJ 14499
# 구현
import copy
n, m, x, y, k = map(int, input().split())
board = []
# 윗 - 뒤 - 오 - 왼 - 앞 - 바닥
dice = [0, 0, 0, 0, 0, 0]

# 동 - 서 - 북 - 남
dx = [0, 0, -1, 1]
dy = [1, -1, 0, 0]

def move_east():
    # 바닥 - 왼 - 윗 - 오 - 바닥
    dice_copy = copy.deepcopy(dice)
    dice[3] = dice_copy[-1]
    dice[0] = dice_copy[3]
    dice[2] = dice_copy[0]
    dice[-1] = dice_copy[2]

def move_west():
    # 바닥 - 오 - 윗 - 왼 - 바닥
    dice_copy = copy.deepcopy(dice)
    dice[2] = dice_copy[-1]
    dice[0] = dice_copy[2]
    dice[3] = dice_copy[0]
    dice[-1] = dice_copy[3]

def move_north():
    # 바닥 - 앞 - 윗 - 뒤 - 바닥
    dice_copy = copy.deepcopy(dice)
    dice[4] = dice_copy[-1]
    dice[0] = dice_copy[4]
    dice[1] = dice_copy[0]
    dice[-1] = dice_copy[1]

def move_south():
    # 바닥 - 뒤 - 윗 - 앞 - 바닥
    dice_copy = copy.deepcopy(dice)
    dice[1] = dice_copy[-1]
    dice[0] = dice_copy[1]
    dice[4] = dice_copy[0]
    dice[-1] = dice_copy[4]

for _ in range(n):
    b = list(map(int, input().split()))
    board.append(b)

moving = list(map(int, input().split()))

nx, ny = x, y
for mv in moving:
    nx += dx[mv-1]
    ny += dy[mv-1]
    if 0 > nx or nx > n-1 or ny < 0 or ny > m-1:
        nx -= dx[mv - 1]
        ny -= dy[mv - 1]
        continue
    if mv == 1:
        move_east()
    elif mv == 2:
        move_west()
    elif mv == 3:
        move_north()
    else:
        move_south()
    if board[nx][ny] == 0:
        board[nx][ny] = dice[-1]
        print(dice[0])
    else:
        dice[-1] = board[nx][ny]
        board[nx][ny] = 0
        print(dice[0])