[BAEKJOON]백준 1018번: 체스판 다시 칠하기 파이썬

문제

지민이는 자신의 저택에서 MN개의 단위 정사각형으로 나누어져 있는 M*N 크기의 보드를 찾았다. 어떤 정사각형은 검은색으로 칠해져 있고, 나머지는 흰색으로 칠해져 있다. 지민이는 이 보드를 잘라서 8*8 크기의 체스판으로 만들려고 한다.

체스판은 검은색과 흰색이 번갈아서 칠해져 있어야 한다. 구체적으로, 각 칸이 검은색과 흰색 중 하나로 색칠되어 있고, 변을 공유하는 두 개의 사각형은 다른 색으로 칠해져 있어야 한다. 따라서 이 정의를 따르면 체스판을 색칠하는 경우는 두 가지뿐이다. 하나는 맨 왼쪽 위 칸이 흰색인 경우, 하나는 검은색인 경우이다.

보드가 체스판처럼 칠해져 있다는 보장이 없어서, 지민이는 8*8 크기의 체스판으로 잘라낸 후에 몇 개의 정사각형을 다시 칠해야겠다고 생각했다. 당연히 8*8 크기는 아무데서나 골라도 된다. 지민이가 다시 칠해야 하는 정사각형의 최소 개수를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 N과 M이 주어진다. N과 M은 8보다 크거나 같고, 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에는 보드의 각 행의 상태가 주어진다. B는 검은색이며, W는 흰색이다.

 

출력

첫째 줄에 지민이가 다시 칠해야 하는 정사각형 개수의 최솟값을 출력한다.

 

코드

N, M = map(int, input().split())
original = []
count = []

for _ in range(N):
    original.append(input())

for a in range(N-7):
    for b in range(M-7):
        index1 = 0
        index2 = 0
        for i in range(a, a+8):
            for j in range(b, b+8):
                if (i+j) % 2 == 0:
                    if original[i][j] != 'W':
                        index1 += 1
                    if original[i][j] != 'B':
                        index2 += 1
                else:
                    if original[i][j] != 'B':
                        index1 += 1
                    if original[i][j] != 'W':
                        index2 += 1
        count.append(min(index1, index2))

print(min(count))

코드 실행 결과

 

코드 설명

N*M 보드를 map() 함수를 이용하여 띄어쓰기로 구분하여 입력 받는다.

원래의 판을 저장하기 위한 리스트 original과 

바뀐 채스판의 갯수를 저장하기 위한 리스트 count를 정의한다.

N(행)의 갯수만큼 원래의 판을 입력 받는다.

original이 리스트 이므로 append(input())을 이용하여 리스트에 추가 해준다.

N, M = map(int, input().split())
original = []
count = []

for _ in range(N):
    original.append(input())

 

가능한 모든 경우의 수를 체크하기 위해 4중 for문을 사용한다.

첫번째, 두번째 for문은 전체 체스판에서 시작점을 잡기 위한 반복문이다.

a는 행, b는 열에 대하여 원래의 체스 판에서 8*8로 자를 수 있는 범위의 시작점을 가리킨다.

index1은 'W'로 시작할 경우 바뀐 체스 판의 갯수를 세기 위함이고,

index2는 'B'로 시작할 경우 바뀐 체스 판의 갯수를 세기 위함이다.

for a in range(N-7):
    for b in range(M-7):
        index1 = 0
        index2 = 0

 

행과 열의 시작점 a, b를 기준으로 8칸씩 모두 체크한다.

현재 행의 번호 i, 현재 열의 번호 j의 합이 짝수이면 시작점의 색깔과 같아야 하고,

홀수이면 시작점의 색깔과 다른 색이어야 한다.

이를 이용하여 만약 (i+j)의 합이 짝수 일 경우,

W가 아니라면 index1에 1을 더하고, B가 아니라면 index2에 1을 더한다.

else문은 (i+j)의 합이 홀수인 경우로, 시작점의 색깔과 다르지 않은 경우를 체크한다.

        for i in range(a, a+8):
            for j in range(b, b+8):
                if (i+j) % 2 == 0:
                    if original[i][j] != 'W':
                        index1 += 1
                    if original[i][j] != 'B':
                        index2 += 1
                else:
                    if original[i][j] != 'B':
                        index1 += 1
                    if original[i][j] != 'W':
                        index2 += 1

 

'W'로 시작할 경우와 'B'로 시작할 경우 바뀐 체스판의 수 중 작은 수를 count 리스트에 더해준다.

모든 경우의 수를 다 체크한 후, count 중 제일 작은 수를 출력하고 프로그램을 종료한다.

        count.append(min(index1, index2))

print(min(count))