글 읽기 - array를 이용한 풀이에서 런타임에러[파이썬]

hanavi91 4년 전 0

런타임에러가 뜹니다.

어디서 뜨는지 모르겠네요.

혹은 반례가 있는 건지 궁금합니다.

import numpy as np

n,m = input().split()
n = int(n)
m = int(m)
bw = []

for k in range(n):
    bw.append(input())
    bw[k]=list(bw[k])

bw = np.array(bw)

def changer(bw, min_ssum=999):
    #체스판 0:'B', 1:'W'
    ans1 = np.array([           # ans1 = 'B'로 시작하는 체스판
        [0,1,0,1,0,1,0,1],      
        [1,0,1,0,1,0,1,0],
        [0,1,0,1,0,1,0,1],
        [1,0,1,0,1,0,1,0],
        [0,1,0,1,0,1,0,1],
        [1,0,1,0,1,0,1,0],
        [0,1,0,1,0,1,0,1],
        [1,0,1,0,1,0,1,0]
    ])
    ans2 = (ans1-1)**2          # ans2 = 'W'로 시작하는 체스판

    bw_rev = (bw == 'W')*1        # 입력된 테이블(bw)을 'W'이면 1 'B'이면 0으로 치환
    
    size_ans_row = size_ans_column = 8
    
    for i in range(len(bw_rev[:,0])-size_ans_column+1):         # 세로 방향으로 몇번 해야하는지
        for j in range(len(bw_rev[0,:])-size_ans_row+1):        # 가로 방향으로 몇번 해야하는지
            #((i,j)에서 8*8행렬 - 체스판)**2의 합은 바꿔칠해야 하는 횟수
            # ans1 체스판 기준 바꿔 칠해야하는 횟수
            sub_ssum1 = bw_rev[i:i+size_ans_column, j:j+size_ans_row]-ans1    
            ssum1 = np.sum(sub_ssum1**2)
            # ans2 체스판 기준 바꿔 칠해야하는 횟수
            sub_ssum2 = bw_rev[i:i+size_ans_column, j:j+size_ans_row]-ans2
            ssum2 = np.sum(sub_ssum2**2)

            # ans1 혹은 ans2 둘 중 한가지에 가까운 녀석으로 선정
            ssum = min(ssum1,ssum2)         

            # 바꿔 칠해야하는 최소값 업데이트
            if (ssum < min_ssum):           
                min_ssum = ssum
                res = min_ssum

    return res

print(changer(bw))

sait2000 4년 전 1

힌트: 1번 줄에서 런타임 에러가 납니다

hanavi91 4년 전 0

첫줄 위치를 바꿔봐도 에러나는데 사용을 하면 안되는 건가요...?

djm03178 4년 전 0

네. numpy는 표준 라이브러리가 아니기 때문에 BOJ에서 지원하지 않습니다.

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

문제

문제

출처

ICPC

array를 이용한 풀이에서 런타임에러[파이썬]

hanavi91 4년 전 0

sait2000 4년 전 1

hanavi91 4년 전 0

djm03178 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말