프로그래밍 문제 풀이/백준(파이썬)

[백준] 1018번: 체스판 다시 칠하기

태은코딩 2024. 3. 23. 18:15

문제

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

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

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

 

입력

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

 

출력

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

 

예제입력1

8 8
WBWBWBWB
BWBWBWBW
WBWBWBWB
BWBBBWBW
WBWBWBWB
BWBWBWBW
WBWBWBWB
BWBWBWBW

예제출력1

1

예제입력2

10 13
BBBBBBBBWBWBW
BBBBBBBBBWBWB
BBBBBBBBWBWBW
BBBBBBBBBWBWB
BBBBBBBBWBWBW
BBBBBBBBBWBWB
BBBBBBBBWBWBW
BBBBBBBBBWBWB
WWWWWWWWWWBWB
WWWWWWWWWWBWB

예제출력2

12

 

문제풀이

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

일단 N과 M을 입력 받는다.

체스판을 입력받을 orginal=[] 리스트를 형성한다.

count는 체스판이 하나는 맨 왼쪽 위 칸이 흰색인 경우, 하나는 검은색인 경우이니까

각 경우의 수에서 다시 칠해야 하는 정사각형의 최소 개수인 경우를 구하기 위해

모든 경우의 수를 일단 count에 append하고 마지막에 min()함수를 통해 최솟값을 출력할 것이다.

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

아까 만들어둔 orginal 리스트에 for문을 활용해 체스판의 각 줄을 입력받는다.

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

문제에서 " 8×8 크기의 체스판으로 잘라낸 후에 몇 개의 정사각형을 다시 칠해야겠다고 생각했다. 당연히 8*8 크기는 아무데서나 골라도 된다." 라고 하였으니

범위가 8×8보다 큰 체스판에서는 맨 왼쪽 위칸을 설정하고 해야한다.

그러니 범위를 N-7, M-7로 잡고 a,b를 맨 왼쪽 위 체스판 좌표로 설정해서 각 8*8체스판 경우의 수를 모두 구해야 한다.

index1은 맨 왼쪽 위가 W인 체스판이라 할때 다시 칠해야 하는 경우의수

index2은 맨 왼쪽 위가 B인 체스판이라 할때 다시 칠해야 하는 경우의수

        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

맨 왼쪽 위 좌표 a,b를 기준으로 for문 범위 range(a,a+8), range(b,b+8)범위로 i와 j를 구해서

i,j 곧 0,0기준 i+j가 짝수인경우와 홀수인 경우로 색이 달라지므로

짝수인 경우 W인지 확인하고 아니면 index1+=1, B인지 확인하고 아니면 index2+=1

홀수인 경우 B인지 확인하고 아니면 index1+=1, W인지 확인하고 아니면 index2+=1

        count.append(min(index1, index2))

print(min(count))

그리고 index1과 index2를 count에 append시킨다. 

이로서 가능한 모든 8*8인 체스판의 왼쪽위가 B거나 W인 경우의 다시 칠해야하는 경우의 수를 구했으니

print(min(count))하면 출력할 수 있다. 

 

정답

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))