글 읽기 - 시간초과 원인 질문입니다[python]

jjhwbe4 3년 전 0

테이스 케이스는 모두 통과하는데, 어디서 시간초과 문제가 발생하는지 원인을 자세히 알려주시면 감사하겠습니다.

단순히 while문 안에 이중for문을 사용했기 때문인가요?

import sys
input = sys.stdin.readline

m,n = map(int, input().split())
pmap = []
#첫쨰날 ~ 끝까지 토마토 계수
pcount = []
count = 0
def mapcount(pmap):
    a= 0
    for i in range(n):
        for j in range(m):
            a += pmap[i][j]
    return a
    
for _ in range(n):
    a = list(map(int,input().split()))
    pmap.append(a)
dx = [1,-1,0,0]
dy = [0,0,1,-1]
def bfs(a,b):
    for i in range(4):
        x = a + dx[i]
        y = b + dy[i]
        if 0<=x<n and 0<=y<m and pmap[x][y] == 0:
            pmap[x][y] = 1
        
pcount.append(mapcount(pmap))
k= 0
q= []
w = []
while True:
    #하루가 지나고
    for i in range(n):
        for j in range(m):
            if pmap[i][j] == 1:
                q.append(i)
                w.append(j)
    for e in range(len(q)):
        bfs(q[e],w[e])
    
    #count += 1
    pcount.append(mapcount(pmap))
    if pcount[k] == pcount[k+1]:
        break
    k += 1
for i in range(n):
    if pmap[i].count(0) != 0:
        k = -1
        break
    else:
        pass     
print(k)
    

    
    
    

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

문제

문제

출처

ICPC

시간초과 원인 질문입니다[python]

jjhwbe4 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말