글 읽기 - python 런타임 에러

quf9484 3년 전 0

런타임 에러가 뜹니다

할당하지 않은 부분을 참조하거나 다른 오류를 체크해봤는데 도저히 모르겠습니다

도와주시면 감사하겠습니다

def print_lab(lab):
    print("")
    for i in range(len(lab)):
        for j in range(len(lab[0])):
            print(lab[i][j], end=" ")
        print("")

from itertools import combinations
import queue
import copy

# 입력
n, m = map(int, input().split())
lab = [[int(x) for x in input().split()] for _ in range(n)]

# 바이러스 위치
virus_pos = []
safe_pos = []
for i in range(n):
    for j in range(m):
        if lab[i][j] == 2:
            virus_pos.append((i, j))
        if lab[i][j] == 0:
            safe_pos.append((i, j))

safe_filed = -1

for (r1,c1),(r2,c2),(r3,c3) in list(combinations(safe_pos, 3)):
    if lab[r1][c1] != 0: continue
    if lab[r2][c2] != 0: continue
    if lab[r3][c3] != 0: continue 

    # 벽 세우기
    tmp_lab = copy.deepcopy(lab)
    tmp_lab[r1][c1] = tmp_lab[r2][c2] = tmp_lab[r3][c3] = 1

    # 바이러스 퍼트리기(bfs)
    dx = [-1, 0, 1, 0]
    dy = [0, -1, 0, 1]
    Q = queue.Queue()
    for pos in virus_pos:
         Q.put(pos)
         while Q.qsize() > 0:
             pos = Q.get()
             for i in range(4):
                    r = dx[i] + pos[0]
                    c = dy[i] + pos[1]
                    if r < 0 or c < 0 or r >= n or c >= m or tmp_lab[r][c] != 0: continue
                    else:
                        Q.put((r, c))
                        tmp_lab[r][c] = 2
    # 0의 개수 체크
    cnt = 0
    for i in tmp_lab:
        cnt += i.count(0)

    if safe_filed < cnt:
        safe_filed = cnt

print(safe_filed)

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

문제

문제

출처

ICPC

python 런타임 에러

quf9484 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말