글 읽기 - 반례를 못찼겠어요

kkjang0717 2년 전 0

2차원 배열에서 x-n, x+n 이런식으로 위아래칸을 확인해주었는데

어디서 틀렸는지 못찾겠네요 ㅜㅜ

도와주세요 ㅜㅜ 웬만한 반례는 다 통과했네요 ㅜㅜ

import sys
from collections import deque

m, n, h = map(int, sys.stdin.readline().split())
tomato = []

for _ in range(n * h):
    tomato.append(list(map(int, sys.stdin.readline().split())))

# tomato[x][y] -> x % n == 0 x+1 %n == 0
q = deque()
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

for i in range(n*h):
    for j in range(m):
        if tomato[i][j] == 1:
            q.append([i, j])

while q:
    x, y = q.popleft()

    for i in range(6):
        if x % n != 0 and (x+1)%n != 0 : #어느칸이든 첫째 막줄 아닐때
            if i < 4:
                nx, ny = x + dx[i], y + dy[i]
            elif i == 4:
                nx, ny = x + n, y
            else:
                nx, ny = x - n, y
            
            if 0 <= nx < n*h and 0 <= ny < m: 
                if tomato[nx][ny] == 0:
                    tomato[nx][ny] = tomato[x][y] + 1
                    q.append([nx, ny])
        elif x % n == 0: #첫줄은 위가 없음 
            # 2차원 배열로 해서 예를들어 높이가 2인칸의 첫줄일때 높이1인 칸의 마지막줄이 바로 윗줄로 되어있어 
            # 그런부분을 피하기 위함
            aa = True
            if i < 4 and dx[i] == -1:#첫째줄은 x값을 빼주면 전칸의 마지막줄이라 안됨 
                aa = False

            if i < 4 and dx[i] != -1:
                nx, ny = x + dx[i], y + dy[i]
            elif i == 4:
                nx, ny = x + n, y
            elif i == 5:
                nx, ny = x - n, y

            if aa:
                if 0 <= nx < n*h and 0 <= ny < m: 
                    if tomato[nx][ny] == 0:
                        tomato[nx][ny] = tomato[x][y] + 1
                        q.append([nx, ny])
        elif (x+1)%n == 0:
            # 마지막 줄일때 예를들어 높이가 2인 칸의 막줄일때 높이 3인칸의 첫줄과 아래위로 붙어있어 
            # 그런부분은 피해줘야함
            bb = True
            if i < 4 and dx[i] == 1: #마지막줄은 x값을 더해주면 다음칸의 첫줄이라 안됨
                bb = False

            if i < 4: # 막줄은 아래가 없음
                if dx[i] != 1:
                    nx, ny = x + dx[i], y + dy[i]
            elif i == 4:
                nx, ny = x + n, y
            else:
                nx, ny = x - n, y

            if bb:
                if 0 <= nx < n*h and 0 <= ny < m: 
                    if tomato[nx][ny] == 0:
                        tomato[nx][ny] = tomato[x][y] + 1
                        q.append([nx, ny])
        
flag = True   
for i in tomato:
    for j in i:
        if j == 0: # 익지않은토마토 남았으면
            flag = False
            break

if flag:
    day = max(tomato)
    print(max(day)-1)   
else:
    print(-1)

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

문제

문제

출처

ICPC

반례를 못찼겠어요

kkjang0717 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말