글 읽기 - 반례나 틀린 부분을 알고싶어요

shhmun02 1년 전 0

질문 게시판에 있는 반례 들을 찾아서 넣어봐도 통과가 나오는데 어느 부분이 잘못된 건지 모르겠습니다.

c, r, h = map(int,input().split()) # 가로 세로 높이
arr = [ [ list(map(int,input().split())) for _ in range(r) ] for _ in range(h) ]

from collections import deque
moving = [[1,0,0],[-1,0,0],[0,1,0],[0,-1,0],[0,0,1],[0,0,-1]]

def bfs(s):
    global day
    spot = deque()
    spot.append(s)
    while spot:
        bowl = []
        site = spot.popleft()
        for s_z, s_y, s_x in site: # 상 하 좌 우 위 아래를 돌며 확인
            for m_z, m_y, m_x in moving:
                site_z = s_z + m_z
                site_y = s_y + m_y
                site_x = s_x + m_x
                if -1 < site_z < h and -1 < site_y < r and -1 < site_x < c:
                    if visited[site_z][site_y][site_x] == False:# 방문하지 않은 경우
                        bowl.append([site_z, site_y, site_x])# 익어 있는 토마토와 인접한 토마토의 좌표
                        visited[site_z][site_y][site_x] = True # 방문 처리
        
        if len(bowl) > 0: # bowl이 0보다 크면 1일 추가
            day += 1
            spot.append(bowl)

#토마토 상자 크기만큼 방문기록 만들기        
visited = [[[False for _ in range(c)] for _ in range(r)] for _ in range(h)] 
day = 0
starts = []
switch = 0
for i in range(h):
    for j in range(r):
        for k in range(c):
            if arr[i][j][k] == 1: # 좌표가 1인 부분을 starts리스트에 추가
                starts.append([i,j,k])
            elif arr[i][j][k] == 0:
                switch = 1
            else:
                visited[i][j][k] = True # arr[i][j][k] 의 값이 -1인 경우 방문 처리
            
# 아래 if문 -> 익은토마토가 존재하고 안익은 토마토도 존재한다.
if switch == 1 and len(starts) > 0: # starts리스트의 크기가 1이상이고 switch == 1 이면 bfs함수 실행
    bfs(starts)
    sw = 0
    for i in visited:
        for j in i:
            if False in set(j):# 익지 않은 토마토 존재시 -1 출력
                print(-1)
                sw = 1
                break
        if sw == 1:
            break
    if sw == 0: # 토마토가 전부 익었으면 익는데 걸린 날을 출력
        print(day)

elif switch == 1 and len(starts) == 0: # 안익은 토마토만 존재시 -1을 출력
    print(-1)
    
else:# 처음부터 전부 익은 토마토만 존재시 0을 출력
    print(0)

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

문제

문제

출처

ICPC

반례나 틀린 부분을 알고싶어요

shhmun02 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말