글 읽기 - [파이썬] 31% 대에서 런타임 에러

alstjgg 3년 전 0

31%대에서 런타임 에러가 발생하는데 원인을 못찾겠습니다ㅠㅠ

도와주세요...

간단하게 함수 설명을 하자면,

count_ice(): 빙산 덩어리가 2개이상이면 True, 그 미만이면 False를 반환하는 함수

> 빙하가 발견되면 que에 집어넣어 bfs 방식으로 덩어리 하나를 발견합니다(visited 배열 이용)

> 그 후에 다시 빙하가 발견된다면, 덩어리가 2개 이상이므로 return True, 끝까지 발견되지 않으면 return False

melt_ice(): 빙하를 녹이는 함수

> 모든 칸을 돌면서 update 배열에 녹이는 양을 저장합니다

> 마지막에 ice 에서 update 값을 빼주어 갱신합니다

check_ice(): 빙하가 존재하는지 검사하는 함수

> 모든 칸을 돌다가 빙하를 발견하면 retrun True, 끝까지 발견하지 못하면 return False

미리 감사드립니다!

from collections import deque
from sys import stdin


def count_ice():
    # TODO: 빙산이 2개 이상의 덩어리로 분리되었는지 검사
    visited = [[False for _ in range(M)] for _ in range(N)]
    que = deque()
    check = False  # 1 덩어리 있으면 True
    for i in range(N):
        for j in range(M):
            if ice[i][j] != 0 and not visited[i][j]:
                if check:  # 기존 덩어리말고 다른 덩어리 발견
                    return True
                visited[i][j] = True
                que.append([i, j])
                while que:
                    cx, cy = que.popleft()
                    for dx, dy in zip([0, 0, 1, -1], [1, -1, 0, 0]):
                        nx, ny = cx + dx, cy + dy
                        if not 0 <= nx < N or not 0 <= ny < M or visited[nx][ny] or ice[nx][ny] == 0:
                            continue
                        visited[nx][ny] = True
                        que.append([nx, ny])
                check = True
    return False


def melt_ice():
    # TODO: 빙하를 녹임
    update = [[0 for _ in range(M)] for _ in range(M)]
    for i in range(N):
        for j in range(M):
            if ice[i][j] == 0:
                continue
            for dx, dy in zip([0, 0, 1, -1], [1, -1, 0, 0]):
                nx, ny = i + dx, j + dy
                if not 0 <= nx < N or not 0 <= ny < M or ice[nx][ny] != 0:
                    continue
                update[i][j] += 1
    for i in range(N):
        for j in range(M):
            if ice[i][j] == 0:
                continue
            ice[i][j] -= update[i][j]
            if ice[i][j] < 0:
                ice[i][j] = 0


def check_ice():
    # TODO: 빙하가 존재하는지 검사
    for i in range(N):
        for j in range(M):
            if ice[i][j] != 0:
                return True
    return False


N, M = map(int, stdin.readline().split())
ice = list(list(map(int, stdin.readline().split())) for _ in range(N))
year = 0
while not count_ice():
    if not check_ice():
        year = 0
        break
    year += 1
    melt_ice()
print(year)

imn00133 3년 전 0

ny, nx가 index를 벗어날 수 있습니다.

alstjgg 3년 전 0

혹시 어디 ny, nx 말씀이신가요??

if not 0 <= nx < N or not 0 <= ny < M

요 부분으로 검출이 안되는 부분이 있는건가요??

imn00133 3년 전 0

아 죄송합니다. 순간적으로 반대로 봤네요.

31번재줄 M M으로 ㅆ싵게 문제인거 같네요.

alstjgg 3년 전 0

.....감사합니다..

허무하네요ㅠㅠ 더 꼼꼼하게 보도록 하겠습니다...

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

문제

문제

출처

ICPC

[파이썬] 31% 대에서 런타임 에러

alstjgg 3년 전 0

imn00133 3년 전 0

alstjgg 3년 전 0

imn00133 3년 전 0

alstjgg 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말