글 읽기 - pypy3는 통과가 돼는데 python3는 1%에서 시간초과가 뜹니다.

rown1424 3년 전 0

질문 게시판 다 뒤져봤는데도 원인을 모르겠습니다ㅠㅠ..

pypy3에선 통과가 되는 걸로 보아 제대로 구현은 한 것 같은데 왜 시간초과가 뜰까요..?

python3에서도 통과 하고 싶은데 도움 좀 부탁드립니다 찝찝해서 잠을 못자고 있습니다...

from collections import deque
m,n,h = map(int,input().split())
a = [[list(map(int,input().split())) for _ in range(n)] for _ in range(h)]
q = deque([])
visited = [[[0] * m for _ in range(n)] for _ in range(h)]
ch = 0
max_num = 0

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

# 토마토가 모두 익지 못하는 경우 체크
def check(a):
    global cnt
    for k in range(h):
        for j in range(n):
            for i in range(m):
                if a[k][j][i] == 0:
                   return -1

# 토마토 시작 위치 담기
for k in range(h):
    for j in range(n):
        for i in range(m):
            if a[k][j][i] == 1:
                q.append((k,j,i))

# BFS
while q:
    sz,sy,sx = q.popleft()
    for s in range(6):
        z = sz + dz[s]
        y = sy + dy[s]
        x = sx + dx[s]
        if 0 <= z < h and 0 <= y < n and 0 <= x < m and visited[z][y][x] == 0 and a[z][y][x] == 0:
            visited[z][y][x] = visited[sz][sy][sx] + 1
            max_num = max(max_num,visited[z][y][x])
            a[z][y][x] = 1
            q.append((z,y,x))

ch = check(a)
if ch == -1:
    print(-1)
else:
    print(max_num)

rlarhkd295 3년 전 0

저도 pypy는 통과하는데 python3는 틀렸는데요

한가지 짚을점이 도움이 되실진 모르겠지만 visited(여기서는 a라는 list)가 필요가 없더군요

어차피 이 코드상의 visited의 값이 0이면 아직 방문하지 않은 점이고 1 이상의 정수면 방문한 셈이니까요

a를 지우고 하셔도 될것같습니다

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

문제

문제

출처

ICPC

pypy3는 통과가 돼는데 python3는 1%에서 시간초과가 뜹니다.

rown1424 3년 전 0

rlarhkd295 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말