글 읽기 - 토마토 런타임 에러 질문입니다.

park3three 3년 전 0

주어진 5개의 테스트케이스는 모두 정답인데 런타임 에러가 발생합니다.

어느 부분이 문제인가요?

import sys
from collections import deque

# 안전지역 체크
def is_safe(y, x):
    return -1 < y < y_row and -1 < x < x_col and visited[y][x] == 0

# BFS
def BFS():

    while Q:
        start_y , start_x = Q.popleft()

        # 익은 토마토 주위 탐색
        for i in range(4):
            new_y = start_y + dy[i]
            new_x = start_x + dx[i]

            if is_safe(new_y, new_x):  # 안전지역이며
                if puzzle[new_y][new_x] >= 0:  # 익은 토마토 주위에 토마토가 있으면
                    puzzle[new_y][new_x] = 1  # 그 토마토를 익히고
                    Q.append((new_y, new_x))  # 큐에 추가
                    visited[new_y][new_x] = visited[start_y][start_x] + 1  # 경과 일 기록


# 상우하좌
dy = [-1, 0, 1, 0]
dx = [0, 1, 0, -1]

x_col, y_row = map(int, sys.stdin.readline().split())
puzzle = [list(map(int, sys.stdin.readline().split())) for _ in range(y_row)]
Q = deque()
visited = [[0]*x_col for _ in range(y_row)]


for i in range(0, y_row):
    for j in range(0, x_col):
        # 이미 익은 토마토들을 큐에 추가
        if puzzle[i][j] == 1:
            Q.append((i, j))
        # 비어 있는 칸은 -1로 표시
        if puzzle[i][j] == -1:
            visited[i][j] = -1

# 이미 익은 토마토들의 좌표를 visited 에 기록
for i in range(0, len(Q)):
    visited[Q[i][0]][Q[i][1]] = 1

BFS()
sub_result = []
flag = 0

for i in range(len(visited)):
    sub_result.append(max(visited[i]))
    for j in range(len(visited)):
        if visited[i][j] == 0:  # 익지 않은 토마토가 남은 상태로 끝난다면
            result = -1
            flag = 1
            break

if not flag:  # 모든 토마토가 익었다면
    result = (max(sub_result) - 1)

print(result)

djm03178 3년 전 1

55번째 줄은 len(visited[i])가 되어야 할 것 같습니다.

park3three 3년 전 0

감사합니다. 해결되었습니다.

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

문제

문제

출처

ICPC

토마토 런타임 에러 질문입니다.

park3three 3년 전 0

djm03178 3년 전 1

park3three 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말