글 읽기 - 3190 뱀 문제에서 제 코드의 오류를 찾는 것이 어렵습니다..

dktkaemfl12 1년 전 0

다른 분들이 올려주신 반례를 사용해본 결과 적은 양의 데이터에서는 정상적인 결과값이 나오지만,

많은 양의 데이터를 입력할 때 오류가 있는 것 같습니다.. 고수님들의 조언 부탁드립니다.

import sys
from collections import deque

input = sys.stdin.readline

N = int(input())

graph = [[0] * (N+1) for _ in range(N+1)]

K = int(input())

for _ in range(K):
    x, y = map(int, input().split())
    graph[x][y] = 1

L = int(input())
direction = []

for _ in range(L):
    count, dir = input().split()
    count = int(count)
    direction.append((count, dir))

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

dir_idx = 0

snake = [1, 1]
count = 0
sx, sy = 1, 1

tail = deque()

while True:
    if len(direction) > 0 and count == direction[0][0]:
        if direction[0][1] == 'L':
            dir_idx = (dir_idx + 3) % 4
        elif direction[0][1] == 'D':
            dir_idx = (dir_idx + 1) % 4
        direction.pop(0)

    nx = sx + dx[dir_idx]
    ny = sy + dy[dir_idx]
    tail.append((sx, sy))

    if nx > N or ny > N or nx < 1 or ny < 1 or (nx, ny) in tail:
        print(count+1)
        exit(0)

    sx = nx
    sy = ny
    count += 1

    if graph[sx][sy] == 1:
        graph[sx][sy] == 0
    else:
        tail.popleft()