글 읽기 - 1799 비숍 문제 valueerror 질문드립니다.

zookeeper_464 2년 전 0

지난번에 못 풀어서 다시 풀어보기 위해 게시판 확인한 뒤 흑백을 나눈 알고리즘을 사용해서 해결하려고 했는데 valueerror가 나왔습니다. 63%에서 해당 error가 나옵니다. 혹시 value error가 나오는 이유에 대해서 알려주실 수 있으신지 궁금합니다.

추가로 주석을 달지 않아서 코드에 대해 말씀드리면 다음과 같습니다.
1. 입력값 입력, 흑백을 구분하여 비숍의 위치 저장

2. 데이터의 위치에 따라서 백트래킹을 활용하여 대각선에 같은 비숍이 있다면 패스, 아니라면 추가하여 탐색하고 마지막에 최종 비숍값 저장하는 함수 저장

3. 최종 값 = 흑비숍 최대 개수 + 백비숍 최대개수

import sys
input = sys.stdin.readline

n = int(input())
chess = [list(map(int,input().split())) for _ in range(n)]
white,black = [], []
for r in range(n):
    for c in range(n):
        if chess[r][c] == 1:
            if (r+c)%2:
                black.append([r,c])
            else:
                white.append([r,c])

def check(idx,rr,ll,cnt,lst):
    global temp
    l = len(lst)

    for i in range(idx+1,l):
        idx1,idx2 = lst[i][0]+lst[i][1],lst[i][0]+(l-1)-lst[i][1]
        if (rr>>idx1)%2 or (ll>>idx2)%2:
            continue

        check(i,rr|(1<<idx1),ll|(1<<idx2),cnt+1,lst)
    
    if cnt>temp:
        temp = cnt

answer = 0
temp = 0
check(-1,0,0,0,white)
answer += temp
temp = 0
check(-1,0,0,0,black)
answer += temp
print(answer)