글 읽기 - (파이썬) 7569 토마토 시간초과 질문있습니다.

shyu0522 2년 전 0

여타 다른 소스들에 비해, for문도 적게돌린다고 생각했는데 계속 시간초과가 납니다.

어느 부분이 문제일까요?

import sys
m, n, h = map(int, sys.stdin.readline().split())
import collections

arr = list()
q = collections.deque()
minus_cnt = 0
for v in range(h):
    temp_arr = list()
    for i in range(n):
        str_case = list(map(int, sys.stdin.readline().split()))
        temp_arr.append(str_case)
        for j in range(len(str_case)):
            if str_case[j] == 1:
                q.append([v,i,j])
            if str_case[j] == -1:
                minus_cnt +=1
    arr.append(temp_arr)
# print(q)
# print(arr)

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

temp = collections.deque()
visited = list()
count=0
cnt_trans = 0
# for i in range(n):
#     for j in range(m):
#         for v in range(h):
#             if arr[v][i][j] == 1:
#                 q.append([v,i,j])

# print(q)
while q:
    z, x, y = q.popleft()
    visited.append([z,x,y])
    for a in range(6):
        nx = x+dx[a]
        my = y+dy[a]
        hz = z+dz[a]
        # print(nx, my)
        if 0<=nx<n and 0<=my<m and 0<=hz<h and arr[hz][nx][my]==0:
            temp.append([hz,nx,my])
            arr[hz][nx][my] = 1
            cnt_trans += 1
    # 한차례 다 돌았을때
    if len(q) == 0:
        # 바꾼놈이 있으면 count++
        if cnt_trans >0:
            count+=1
        q = temp.copy()
        cnt_trans=0
        temp.clear()

if len(visited)!=(m*n*h)- minus_cnt:
    count=-1
# print(len(visited), minus_cnt)
# for i in range(n):
#     for j in range(m):
#         for v in range(z):
#             if arr[v][i][j]==0:
#                 count=-1
print(count)

# 4 3 2
# 1 1 1 1
# 1 1 1 1
# 1 1 1 1
# 1 1 1 1
# -1 -1 -1 -1
# 1 1 1 -1

djm03178 2년 전 1

PyPy3로 제출하면 여유있게 통과됩니다.

shyu0522 2년 전 0

선생님 제가 pypy3를 잘 몰라서 그러는데, 파이썬이랑 크게 다른가요?

djm03178 2년 전 2

기본적으로 둘 다 똑같이 Python 3에 대한 인터프리터입니다. PyPy3가 일반적으로 대부분의 경우 Python 3보다 훨씬 좋은 성능을 내기 때문에 앞으로도 항상 PyPy3만 쓰시는 것이 좋습니다.

shyu0522 2년 전 0

감사합니다 선생님

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

문제

문제

출처

ICPC

(파이썬) 7569 토마토 시간초과 질문있습니다.

shyu0522 2년 전 0

djm03178 2년 전 1

shyu0522 2년 전 0

djm03178 2년 전 2

shyu0522 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말