[Python] 뱀

문제

3190번: 뱀

입력

• 첫째 줄에 보드의 크기 N이 주어진다. (2 ≤ N ≤ 100) 다음 줄에 사과의 개수 K가 주어진다. (0 ≤ K ≤ 100)
• 다음 K개의 줄에는 사과의 위치가 주어지는데, 첫 번째 정수는 행, 두 번째 정수는 열 위치를 의미한다. 사과의 위치는 모두 다르며, 맨 위 맨 좌측 (1행 1열) 에는 사과가 없다.
• 다음 줄에는 뱀의 방향 변환 횟수 L 이 주어진다. (1 ≤ L ≤ 100)
• 다음 L개의 줄에는 뱀의 방향 변환 정보가 주어지는데,  정수 X와 문자 C로 이루어져 있으며. 게임 시작 시간으로부터 X초가 끝난 뒤에 왼쪽(C가 'L') 또는 오른쪽(C가 'D')로 90도 방향을 회전시킨다는 뜻이다. X는 10,000 이하의 양의 정수이며, 방향 전환 정보는 X가 증가하는 순으로 주어진다.

출력

• 첫째 줄에 게임이 몇 초에 끝나는지 출력한다.

문제 풀이

• 방향을 동남서북으로 저장하면 1을 더해서 오른쪽 회전 및 1을 빼서 왼쪽 회전 가능하다
• 사과가 있는 곳의 값을 1, 뱀의 몸통과 머리가 있는 곳의 값을 2로 저장
• 뱀의 위치를 쉽게 알기 위해 snake튜플을 따로 만들어 뱀의 머리부터 꼬리까지의 정보를 저장한다
• 사과가 있는 경우, 이동할 곳의 값을 2로 변경한 후, 뱀의 위치를 저장하는 튜플에 위치 추가
• 사과가 없는 경우, 이동할 곳의 값을 2로 변경한 후, 뱀의 위치 추가 및 뱀의 위치에서 꼬리 제거(가장 먼저 저장 된 값) 후, 꼬리가 있었던 보드의 값을 0으로 변경
• 다음 탐색을 위해 현재의 위치를 이동한 곳의 위치로 업데이트 한 후, 소요시간도 증가시켜준다
• 그 후 방향 전환 해야 하는 경우 방향 전환 후, 방향 전환 리스트에서 처리한 값 삭제
• 뱀이 보드에서 벗어나거나, 자기 몸에 부딪히는 경우 시간 한 번 더 카운트 한 후 소요 시간 출력

코드

n = int(input())
board = [[0] * n for _ in range(n)]
# 사과 개수
k = int(input())
# 사과 위치
for _ in range(k):
    i, j = map(int,input().split())
    board[i - 1][j - 1] = 1
# 뱀의 방향 변환 횟수
l = int(input())
# 뱀의 방향 변환 정보
rotate = []
for _ in range(l):
    x,c = input().split()
    rotate.append((int(x),c))

# 방향전환 (동남서북)
dy = [0, 1, 0, -1]
dx = [1, 0, -1, 0]

# 뱀 움직이기
def simulate():
    # 시작위치
    i, j = 0, 0
    # 뱀의 위치 저장
    board[i][j] = 2
    # 이동 시간
    time = 0
    # 뱀의 방향
    dir = 0
    # 뱀의 몸 위치 저장
    snake = []
    snake.append((0,0))
    # 뱀 이동시키기
    while True:
        new_y = i + dy[dir]
        new_x = j + dx[dir]
        # 뱀이 보드에서 벗어나거나 자기 몸에 부딪히는지 확인
        if 0 > new_x or 0 > new_y or new_x >= n or new_y >= n or board[new_y][new_x] == 2:
            time += 1
            break

        # 다음 위치로 이동 가능
        else:
            # 다음 위치가 사과일 때 (사과 없애고 꼬리 움직이지 x)
            if board[new_y][new_x] == 1:
                board[new_y][new_x] = 2
                # 뱀 머리 위치 저장
                snake.append((new_y, new_x))
            # 사과 아닌 경우
            else:
                board[new_y][new_x] = 2
                snake.append((new_y, new_x))
                # 뱀 꼬리 제거
                py, px = snake.pop(0)
                board[py][px] = 0

        # 현재 위치 변경
        i, j = new_y, new_x
        # 시간 증가
        time += 1

        # 방향 해야하는 경우
        if len(rotate) >= 1 and time == rotate[0][0]:
            # 오른쪽 방향 전환
            if rotate[0][1] == 'D':
                dir = (dir + 1) % 4
            # 왼쪽 방향 전환
            else:
                dir = (dir - 1) % 4
            # 방향 리스트에서 제거
            rotate.pop(0)
    return time
print(simulate())