글 읽기 - 3차원만 정답이 되고 2차원은 틀립니다..

lbg030 1년 전 0

파이썬 뉴비 입니다

토마토 문제를 푸는데 3차원에서는 정상적으로 맞다고 뜨는데 2차원으로 풀어보았을 때 제 생각에는 정상적으로 작동해야 된다고 생각하는데 자꾸 틀렸다고 하고 반례를 돌려봤는데도 다 맞게나와서... 왜 그런지 찾고 싶습니다 ㅜㅜ

from sys import stdin
from collections import deque
n, m, h = map(int, input().split())

lst = [list(map(int, stdin.readline().rstrip().split())) for _ in range(m*h)]

cnt = 0
templist = deque()
temp = []


dx = [0, 0, 1, -1]
dy = [1, -1, 0, 0]
for i in range(m*h):
    for j in range(n):
        if lst[i][j] == 1:
            temp.append((i, j))
    
templist.append(temp)


while templist :
    temp = []
    cnt += 1
    k = templist.popleft()
    for i in k:
        x,y = i
        
        #px mx는 상 하를 계산함.
        px = x + m
        mx = x - m
        
        if 0 <= px < m*h:
            if lst[px][y] == 0:
                lst[px][y] = 1
                temp.append((px,y))
                
        if 0 <= mx < m*h:
            if lst[mx][y] == 0:
                lst[mx][y] = 1
                temp.append((mx,y))
                
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0 <= nx < m*h and 0 <= ny < n:
                if lst[nx][ny] == 0:
                    lst[nx][ny] = 1
                    temp.append((nx, ny))
                else :
                    continue
                
    if temp:
        templist.append(temp)
        
    # print(cnt)
    # print(lst)

flag = 1

for x in lst:
    if x.count(0) > 0 :
        flag = 0

print("-1" if flag == 0 else cnt-1)

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

문제

문제

출처

ICPC

3차원만 정답이 되고 2차원은 틀립니다..

lbg030 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말