글 읽기 - [파이썬] 시간초과 해결방법

tkddnjs1234 3년 전 0

기본적인 bfs 방식으로 풀었는데

pypy로는 정답이 뜨는데 python으로 채점하면 시간초과가 뜨네요...

더 이상 어디서 시간을 줄여야할지 몰라서 알려주시면 감사하겠습니다!!!

# https://www.acmicpc.net/problem/7569
from collections import deque

m,n,h = list(map(int, input().split()))
arr = []
for _ in range(h):
    temp = []
    for _ in range(n):
        temp.append(list(map(int, input().split())))
    arr.append(temp)

q = deque()
visit = set()
for i in range(h):
    for j in range(n):
        for k in range(m):
            if arr[i][j][k] == 1:
                q.append((i,j,k))
                visit.add((i,j,k))

direction = ((1,0,0),(-1,0,0),(0,1,0),(0,-1,0),(0,0,1),(0,0,-1))
l = len(q)
day = 0
while q:
    for _ in range(l):
        x,y,z = q.popleft()
        arr[x][y][z] = 1
        for dx,dy,dz in direction:
            nx, ny, nz = x+dx, y+dy, z+dz
            if 0<=nx<h and 0<=ny<n and 0<=nz<m:
                if arr[nx][ny][nz] == 0 and (nx,ny,nz) not in visit:
                    q.append((nx,ny,nz))
                    visit.add((nx,ny,nz))
    day += 1
    l = len(q)

def main(arr, day, h, n):
    for i in range(h):
        for j in range(n):
            if 0 in arr[i][j]:
                print(-1)
                return
    print(day-1)

main(arr, day, h, n)

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

문제

문제

출처

ICPC

[파이썬] 시간초과 해결방법

tkddnjs1234 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말