baekjoon. 카드2 [python] [Silver IV]

문제링크 :https://www.acmicpc.net/problem/2164

2164번: 카드2

 

from collections import deque


N = int(input()) # 보드의 크기 
K = int(input()) # 사과의 개수 

def printAll(board):
    for i in range(N):
        for j in range(N):
            print(board[i][j], end = ' ')
        print("")
    print("")

# 사과 갯수 
# 이처럼 하면 한 행만 바꾸어도 다른 행이 영향 받음 board = [[0] * N] * N
board =  [[0] * N for _ in range(N)]
for _ in range(K):
    appleX, appleY = map(int, input().split())
    board[appleX - 1][appleY - 1] = 'A'

# debug - printAll(board)

L = int(input()) # 방향 번환 횟수 
runs = {}
for _ in range(L):
    X, C = input().split()
    # runs[X] = C;
    runs[int(X)] = C;


# right, down, left, up
dx = [0, 1, 0, -1]
dy = [1, 0, -1, 0]



directions = {'right': 0, 'down': 1, 'left': 2, 'up': 3}
currentDirection = 'right'
currentX = 0
currentY = 0
board[currentX][currentY] = 'S'


def nextDirection(currentDirection, next):
    # print('next - ', currentDirection, ' : ', next)

    if (currentDirection == 'right' and next == 'L'):
        return 'up'
    elif (currentDirection == 'right' and next == 'D'):
        return 'down'
    elif (currentDirection == 'left' and next == 'L'):
        return 'down'
    elif (currentDirection == 'left' and next == 'D'):
        return 'up'
    elif (currentDirection == 'up' and next == 'L'):
        return 'left'
    elif (currentDirection == 'up' and next == 'D'):
        return 'right'
    elif (currentDirection == 'down' and next == 'L'):
        return 'right'
    elif (currentDirection == 'down' and next == 'D'):
        return 'left'
    else:
        return ''
    
time = 0
# 배열로 저장 snake = deque([currentX, currentY])
snake = deque([[currentX, currentY]])
while True:
    # printAll(board)

    # print(runs.keys())
    # print('time - ', time, ' : currentDirection', currentDirection)

    nextX = currentX + dx[directions[currentDirection]]
    nextY = currentY + dy[directions[currentDirection]]

    # print(nextX, " : ", nextY)

    if (nextX < 0 or nextX >= N or nextY < 0 or nextY >= N):
        break;
    if (board[nextX][nextY] == 'S'):
        break;

    # 다음이 사과가 아니면, 큐에서 하나 삭제하고 새로운 좌표를 넣는다. 
    if (board[nextX][nextY] == 0):
        delX, delY = snake.popleft()
        # print('del', delX, ' : ', delY)
        board[delX][delY] = 0 

    snake.append([nextX, nextY])
    board[nextX][nextY] = 'S'

    currentX = nextX
    currentY = nextY
    time = time + 1

    if (time in runs.keys()):
        currentDirection = nextDirection(currentDirection, runs[time])




print(time + 1)