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

[Python] 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())
chs = [input() for _ in range(n)]
t1 = 'WBWBWBWB'
t2 = 'BWBWBWBW'
ans1 = [t1, t2, t1, t2, t1, t2, t1, t2]
ans2 = [t2, t1, t2, t1, t2, t1, t2, t1]
ans = []
tmp = []      
for i in range(n - 7):
    for j in range(m - 7):
        cnt = 0
        for k in range(8):
            for l in range(8):
                if chs[i+k][j+l] != ans1[k][l]:
                    cnt += 1
        tmp.append(cnt)
        cnt = 0
        for k in range(8):
            for l in range(8):
                if chs[i+k][j+l] != ans2[k][l]:
                    cnt += 1
        tmp.append(cnt)

print(min(tmp))

 

주의사항 (개인적으로 문제를 제대로 읽지 않아서 시간 잡아먹었던 사항)

1. 체스판을 색칠하는 경우는 두 가지뿐이다. 하나는 맨 왼쪽 위 칸이 흰색인 경우, 하나는 검은색인 경우이다.

2. 당연히 8*8 크기는 아무데서나 골라도 된다

 

바보같이 입력 리스트 [0][0] 이 W 이냐 B 이냐에 따라 거기에 맞는 맵을 만들고

그 맵을 비교하는 이상한 행동을 했다...  문제 이해력이 안좋은듯 하다...

 

t1 = 'WBWBWBWB'
t2 = 'BWBWBWBW'
ans1 = [t1, t2, t1, t2, t1, t2, t1, t2]
ans2 = [t2, t1, t2, t1, t2, t1, t2, t1]

처음 시작 이 W 인 경우와 B 인 경우에 대해 미리 리스트를 만들었다

 

for i in range(n - 7):
    for j in range(m - 7):

비교할 대상(미리 만들어놓은 리스트) 크기가 8 * 8 이므로

입력받는 가로 세로 크기에서 -7을 하였다

-8을 하면 가로 세로 8 * 8 인 입력이 들어오면 반복문 자체를 실행할수 없어서

[0][0] 지점에서 한번 실행할수있게 -7 을 해준다

이미지의 숫자는 상관 없고 대충 노란 범위가 이동하며 비교하는 느낌 으로 생각하면 된다

 

        cnt = 0
        for k in range(8):
            for l in range(8):
                if chs[i+k][j+l] != ans1[k][l]:
                    cnt += 1                    
        tmp.append(cnt)
        
        cnt = 0
        for k in range(8):
            for l in range(8):
                if chs[i+k][j+l] != ans2[k][l]:
                    cnt += 1
        tmp.append(cnt)

반으로 나눠서 위에 부분은 좌측 상단 시작이 W 인 경우,

아래는 B 로 시작하는 경우를 나눠서 반복문을 사용하였다

 

if chs[i+k][j+l] != ans1[k][l]:

입력한 chs 리스트는 위에 이미지 처럼 부분 부분을 대상으로 해야하기 때문에

이동한 위치(i) + k (0 ~ 7), j + l (0 ~ 7) 로 계산하였고

이미 생성한 ans1 과 ans2 는 크기가 고정이고 range(8) 로 비교하면 되기에 [k][l] 해주었다

 

각각 비교하여 같지 않으면 카운트 +1 씩 하여 tmp 리스트에 append() 한다

 

print(min(tmp))

현재 tmp 리스트에는 시작이 W 인경우의 일치하지 않는 카운트 수 와

시작이 B 인경우의 일치하지 않는 카운트 수가 함께 있기에

그중에 가장 작은 수 min() 으로 출력해준다

 

 

 

이미지 출처

https://gruuuuu.github.io/machine-learning/cnn-doc/