글 읽기 - python 시간초과(pypy3)

susia1005 2년 전 0

계속 시간초과가 나서, 구글에 있는 통과한 파이썬 코드 복붙했는데도 시간초과가 뜨네요. 왜 그럴까요?

(게시글에 있는 코드는 제가 작성한 코드입니다.)

# 시간초과..?
import sys
from collections import deque


def bfs():
    queue = deque()
    # 3차원 배열 선언
    visited = [[[0]*(K+1) for _ in range(M)] for _ in range(N)]
    queue.append((0, 0, 0))
    visited[0][0][0] = 1

    while queue:
        cr, cc, w = queue.popleft()
        if cr == N-1 and cc == M-1:
            print(visited[cr][cc][w])
            return
        for dr, dc in [(-1, 0), (1, 0), (0, -1), (0, 1)]:
            nr, nc = cr + dr, cc + dc
            if 0 <= nr < N and 0 <= nc < M and not visited[nr][nc][w]:
                # 벽이 없는 경우
                if arr[nr][nc] == 0:
                    visited[nr][nc][w] = visited[cr][cc][w] + 1
                    queue.append((nr, nc, w))
                # 벽이 있는 경우
                elif arr[nr][nc] == 1 and w < K:
                    visited[nr][nc][w+1] = visited[cr][cc][w] + 1
                    queue.append((nr, nc, w+1))
    return -1


N, M, K = map(int, sys.stdin.readline().split())
arr = [list(map(int, sys.stdin.readline().rstrip())) for _ in range(N)]

bfs()

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

python 시간초과(pypy3)

susia1005 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말