인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

인프런 커뮤니티 질문&답변

이경훈님의 프로필 이미지
이경훈

작성한 질문수

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비)

15. 토마토(BFS : Breadth First Search)

4,5번 시간초과

작성

·

205

0

import queue

dr = [0,-1,0,1]
dc = [-1,0,1,0]

m,n = map(int,input().split())
arr = [list(map(int,input().split())) for _ in range(n)]
q = queue.Queue()
res = 0

for i in range(n):
    for j in range(m):
        if arr[i][j] == 1:
            q.put((i,j))

while q.qsize()>0:
    now = q.get()
    for k in range(4):
        nr = now[0]+dr[k]
        nc = now[1]+dc[k]
        if 0<= nr < n and 0<= nc < m and arr[nr][nc] == 0:
            arr[nr][nc] = arr[now[0]][now[1]] + 1
            q.put((nr,nc))

c = False
for i in range(n):
    for j in range(m):
        if arr[i][j] == 0:
            c = True
            break;
        else:
            res = max(res, arr[i][j])

if c == True:
    print(-1)
else:
    print(res-1)

4번 5번에서 자꾸 시간초과가 뜨는데 컴퓨터 사양 문제일까요?

직접 cmd에서 구동시켜보면 답은 맞습니다..

답변 1

0

김태원님의 프로필 이미지
김태원
지식공유자

queue 대신  deque로 해보시기 그래도 시간초과가 나면 채점하는 컴퓨터 성능의 문제라 생각됩니다.

이경훈님의 프로필 이미지
이경훈

작성한 질문수

질문하기