[solved.ac 실버5] 체스판 다시 칠하기

제약 조건

문제

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

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

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

입력

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

출력

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

예제

---

나의 풀이

이 문제의 유형은 브루트포스 알고리즘이다. 무차별 대입하여 풀이하는 문제라는 것이다.

즉 문제에서처럼 8X8 크기의 데이터를 비교해가며 변화할 점이 가장 적은 값을 구하면 되는 문제이다.

이때 문제에서 힌트를 준 것처럼 왼쪽 위가 검정으로 시작하는 경우와 흰색으로 시작하는 경우 두 가지가 존재하기 때문에 두 경우 모두 구해서 그 중 더 작은 값을 선택해야 한다.

def countdiff(str1, str2):
    answer = 0
    for i in range(len(str1)):
        if str1[i]!=str2[i]: answer+=1
    return answer

black = ["BWBWBWBW","WBWBWBWB"]
white = ["WBWBWBWB","BWBWBWBW"]
n, m = map(int, input().split())
board = [input() for _ in range(n)]
answer = 64
for startRow in range(n-7):
    for col in range(m-7):
        blackDiff, whiteDiff = 0, 0
        for row in range(8):
            blackDiff += countdiff(board[startRow+row][col:col+8], black[row%2])
            whiteDiff += countdiff(board[startRow+row][col:col+8], white[row%2])
        answer = min(answer, blackDiff, whiteDiff)
print(answer)

시작할 열을 0~n-8까지 반복한다. n-8까지 반복하는 이유는 시작 열에서 8개의 열을 확인하기 때문에 열 길이 - 8까지만 반복하는 것이다.

그리고 col도 마찬가지로 0~m-8까지 반복한다. 열과 같은 이유이다.

길이가 같은 두 문자열에서 같은 위치에서 다른 문자가 몇 개인지 카운트하는 함수 countdiff를 선언했다.

그리고 board[startRow+row][col:col+8] 과 black[row%2] 를 countdiff로 비교했다.

마찬가지로 board[startRow+row][col:col+8] 과 white[row%2] 를 countdiff로 비교했다.

반복문을 돌고 나면 blackDiff에는 검정으로 시작하는 보드와 다른 칸 개수, whiteDiff에는 하양으로 시작하는 보드와 다른 칸 개수가 저장된다.

그 중 더 작은 값을 찾아 나가면 된다.

 

채점 결과