백준 알고리즘 3190번 뱀 풀이 With Python

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

3190번: 뱀

문제는 아래와 같습니다.

문제의 핵심은 뱀이 이동하는 것을 어떻게 구현하는 것인가 하는 것이다. 나는 큐를 사용하여 구현하였는데, 다른 사람들의 코드를 보니 굳이 큐를 사용하지 않고 리스트만으로도 구현할 수 있었다. 내가 해결한 코드는 다음과 같다.

from collections import deque


def solution(map1, dirc, n):
    snake = deque()
    snake.append([0, 0])
    dx = [-1, 0, 1, 0]
    dy = [0, -1, 0, 1]
    answer = 0
    d = 3
    x1, y1 = (0, 0)
    while 1:
        endflag=0
        flag = 0
        answer += 1
        temp = deque()
        for i in range(len(snake)):
            if i == 0:
                nx = snake[0][0]
                ny = snake[0][1]
                temp.append([nx, ny])
                nx += dx[d]
                ny += dy[d]
                if -1 < nx < n and -1 < ny < n and [nx, ny] not in snake:
                    if map1[nx][ny] == 1:
                        flag = 1
                        map1[nx][ny] = 0
                    snake[0] = [nx, ny]
                else:
                    endflag=1
                    break

            else:
                temp.append(snake[i])
                snake[i] = temp.popleft()
        if endflag==1:
            break
        if flag == 1:
            snake.append(temp.pop())

        if dirc and answer == dirc[0][0]:
            if dirc[0][1] == 'D':
                d = (d - 1) % 4
            else:
                d = (d + 1) % 4
            dirc.popleft()
    return answer


n = int(input())
map1 = [[0 for _ in range(n)] for _ in range(n)]
k = int(input())
for _ in range(k):
    a, b = map(int, input().split())
    map1[a - 1][b - 1] = 1
l = int(input())
dirc = deque()
for _ in range(l):
    a, b = input().split()
    a = int(a)
    dirc.append([a, b])
answer=solution(map1,dirc,n)
print(answer)