[백준] 14499 주사위 굴리기(python 파이썬)

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

14499번: 주사위 굴리기

풀이 과정

가장 중요한 포인트는 주사위 6면의 정보 저장방법이다.

구조체나 클래스를 생각했지만 그냥 리스트로 해도 될거같아서 6크기의 리스트 선언하여 해결함.

이동은 간단한 함수 4개를 만들어서 주사위 리스트의 값들을 교환 시켰다.

temp 임시 변수를 하나 선언해서 쉽게 교환 가능.

import sys
input = sys.stdin.readline
N, M, X, Y, K = map(int,input().rstrip().split())

dx = [0,0,-1,1]
dy = [1,-1,0,0] # 1 동, 2 서, 3 북, 4 남
dice = [0,0,0,0,0,0] # down left front back right up
road = []

for i in range(N):
    road.append([int(x) for x in input().rstrip().split()])
command = [int(x) for x in input().rstrip().split()]

상하좌우 입력하기 편하게 dx, dy를 선언하고 주사위의 6면의 값을 저장하기 위해 dice리스트를 선언해준다.

 

def move_right():
    temp = dice[1]
    dice[1] = dice[0]
    dice[0] = dice[4]
    dice[4] = dice[5]
    dice[5] = temp

이동은 함수를 만들어서 표현하였다. 오른쪽으로 주사위가 이동하면 위치가 바뀌기 때문에 (예: 위 -> 오른쪽)

간단하게 값을 swap해주는 함수를 만들어 사용. 다른 방향도 마찬가지로 작성하면 된다.

 

 

x, y = X, Y
for i in command:
    if 0 <= x + dx[i - 1] < N and 0 <= y + dy[i - 1] < M:
        x, y = x + dx[i - 1], y + dy[i - 1]
        if i == 1:
            move_right()
        elif i == 2:
            move_left()
        elif i == 3:
            move_back()
        elif i == 4:
            move_front()

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

주어진 명령에 따라 주사위를 이동시켜주면 된다. 가장 먼저 주사위가 이동 가능한지 판별하고, 

이동 방향에 따라 4가지 함수를 실행시켜준다.

만약 해당 위치의 지도값이 0이면 지도에 주사위바닥의 값을 입력하고 아니면 주사위에 지도값을 입력한다.

마지막으로 주사위 윗면의 값을 출력하면 된다.

 

아래는 전체코드

import sys
input = sys.stdin.readline
N, M, X, Y, K = map(int,input().rstrip().split())

dx = [0,0,-1,1]
dy = [1,-1,0,0] # 1 동, 2 서, 3 북, 4 남
dice = [0,0,0,0,0,0] # down left front back right up
road = []

for i in range(N):
    road.append([int(x) for x in input().rstrip().split()])
command = [int(x) for x in input().rstrip().split()]

#print(road)

def move_right():
    temp = dice[1]
    dice[1] = dice[0]
    dice[0] = dice[4]
    dice[4] = dice[5]
    dice[5] = temp

def move_left():
    temp = dice[4]
    dice[4] = dice[0]
    dice[0] = dice[1]
    dice[1] = dice[5]
    dice[5] = temp

def move_front():
    temp = dice[3]
    dice[3] = dice[0]
    dice[0] = dice[2]
    dice[2] = dice[5]
    dice[5] = temp

def move_back():
    temp = dice[5]
    dice[5] = dice[2]
    dice[2] = dice[0]
    dice[0] = dice[3]
    dice[3] = temp

x, y = X, Y
for i in command:
    if 0 <= x + dx[i - 1] < N and 0 <= y + dy[i - 1] < M:
        x, y = x + dx[i - 1], y + dy[i - 1]
        if i == 1:
            move_right()
        elif i == 2:
            move_left()
        elif i == 3:
            move_back()
        elif i == 4:
            move_front()

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