[백준🏅4] 14499 - 주사위 굴리기 (Python)

Review

주사위라는 단어에 쫄았다.
문제를 풀면서 중요한건 "규칙"을 찾으려고 노력해야 한다는 것
머릿속으로 그려봤을 때 안된다 하더라도,
바로 남의 해설을 보는 것보다,
손으로도 써보고 혼자 규칙을 찾으려고 노력하는 시간이 훨씬 값지다!

✨ HOW? 처음 아이디어

일단 방향이 나오니, dx/dy를 떠올렸다.

아, 그리고 dx/dy 설정할 때 아무렇게 두는 게 아니라
조건에서 요구하는 사항대로 맞추자.

이 문제에선 동서북남이니,

dx = [0, 0, -1, 1]
dy = [1, -1, 0, 0]

✨ HOW? 주사위 규칙

주사위의 면이 [1,2,3,4,5,6] 있을 때 [위, 뒤, 오른, 왼, 앞, 바닥] 이라는 기준을 두고,
"동서남북 각 방향으로 굴렸을 때 이 인덱스들이 어떻게 변하나"만 찾았으면 됐다.

Code

'''
주사위는 지도 위에 윗 면이 1이고, 동쪽을 바라보는 방향이 3인 상태로 놓여져 있으며,  > ???
놓여져 있는 곳의 좌표는 (x, y) 이다. 가장 처음에 주사위에는 모든 면에 0이 적혀져 있다. > 그래서 주사위는 6면이 다 0이라는 말??

지도의 각 칸에는 정수가 하나씩 쓰여져 있다. 
주사위를 굴렸을 때, 이동한 칸에 쓰여 있는 수가 0이면, 
주사위의 바닥면에 쓰여 있는 수가 칸에 복사된다. 

0이 아닌 경우에는, 칸에 쓰여 있는 수가 주사위의 바닥면으로 복사되며, 
칸에 쓰여 있는 수는 0이 된다.

주사위를 놓은 곳의 좌표와 이동시키는 명령이 주어졌을 때,
주사위가 이동했을 때 마다 상단에 쓰여 있는 값을 구하는 프로그램을 작성하시오. > ???

주사위는 지도의 바깥으로 이동시킬 수 없다. 
만약 바깥으로 이동시키려고 하는 경우에는 해당 명령을 무시해야 하며, 출력도 하면 안 된다.
'''

'''첫째 줄에 지도의 세로 크기 N, 가로 크기 M (1 ≤ N, M ≤ 20), 
주사위를 놓은 곳의 좌표 x, y(0 ≤ x ≤ N-1, 0 ≤ y ≤ M-1), 
그리고 명령의 개수 K (1 ≤ K ≤ 1,000)가 주어진다.
동쪽은 1, 서쪽은 2, 북쪽은 3, 남쪽은 4로 주어진다
이동할 때마다 주사위의 윗 면에 쓰여 있는 수를 출력한다.'''

''' 방향이 있으니 dx dy'''

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

'''동서북남 으로 방향 설정'''
dx = [0, 0, -1, 1]
dy = [1, -1, 0, 0]
dice = [0, 0, 0, 0, 0, 0]

def turn(direction):
    a, b, c, d, e, f = dice[0], dice[1], dice[2], dice[3], dice[4], dice[5]
    if direction == 1:
        dice[0], dice[1], dice[2], dice[3], dice[4], dice[5] = d, b, a, f, e, c
        
    elif direction == 2:
        dice[0], dice[1], dice[2], dice[3], dice[4], dice[5] = c, b, f, a, e, d
        
    elif direction == 3:
        dice[0], dice[1], dice[2], dice[3], dice[4], dice[5] = e, a, c, d, f, b
        
    elif direction == 4:
        dice[0], dice[1], dice[2], dice[3], dice[4], dice[5] = b, f, c, d, a, e
    
for d in directions:
    if 0 <= x + dx[d-1] < n and 0 <= y + dy[d-1] < m:
        x = x + dx[d-1]
        y = y + dy[d-1]
        turn(d)
        
        if arr[x][y] == 0:
            arr[x][y] = dice[5] 
        
        else:
            dice[5] = arr[x][y]
            arr[x][y] = 0
        
        print(dice[0])

시간복잡도

O(n)