글 읽기 - [토마토 7569] 파이썬 반례 부탁드립니다.

scleev 2년 전 0

질문게시판에 있는 반례 웬만한 건 다 입력해봤는데 다 통과합니다...

채점 시에는 20% 정도에서 틀렸습니다가 뜹니다.

PT()는 디버깅용 토마토 상자 상태 출력해주는 함수입니다.

from collections import deque
from sys import stdin
input = stdin.readline

M, N, H = map(int, input().split())
tomatoes = []

tomatoes = [[list(map(int, input().split())) for _ in range(N)] for _ in range(H)]
direct = [(1, 0, 0), (-1, 0, 0), (0, 1, 0), (0, -1, 0), (0, 0, 1), (0, 0, -1)]

#Debugging
def PT():
    for k in range(H):
        for i in range(N):
            for j in range(M):
                print(tomatoes[k][i][j], end = ' ')
            print()
    print()

queue  = deque([])
#visited = [[False for _ in range(M)] for _ in range(N)]

#initialization
for k in range(H):
    for i in range(N):
        for j in range(M):
            if tomatoes[k][i][j] == 1:
                queue.append((k, i, j))
    
while queue:

    PT()

    (z, x, y) = queue.popleft()
        
    for (dz, dx, dy) in direct:
        (nz, nx, ny) = (z + dz, x + dx, y + dy)
        if 0 <= nz < H and 0 <= nx < N and 0 <= ny < M and tomatoes[nz][nx][ny] == 0:
            queue.append((nz, nx, ny))
            tomatoes[nz][nx][ny] = tomatoes[z][x][y] + 1

failed = False
result = 0
for k in range(H):
    for i in range(N):
        for j in range(M):
            if tomatoes[k][i][j] == 0:
                failed = True
                result = 0
                break
            result = max(result, tomatoes[k][i][j])
        if failed:
            break

print(result - 1)

sjin0704 2년 전 0

마지막에 결과를 구하는 반복문의 가장 바깥 루프를 탈출하는 코드가 빠졌네요.

scleev 2년 전 0

감사합니다!

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

문제

문제

출처

ICPC

[토마토 7569] 파이썬 반례 부탁드립니다.

scleev 2년 전 0

sjin0704 2년 전 0

scleev 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말