백준 1018 파이썬 답안

문제

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

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

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

입력

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

출력

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

브루트포스로 접근

: 이 문제는 브루트포스로 접근하는 방법을 연습하기 위해 브루트포스로 만들었다. 브루트포스로 접근할 경우, 총 2가지 경우를 고려해야한다. 가장 첫번째 시작하는 블럭이 'W'로 시작할지, 'B로' 시작할지로 두 가지 경우가 나누어진다. 시간 절약을 위해서 완전 탐색 시, 한번에 이것을 두 개 다 확인하도록 접근해야한다.

나같은 경우에는 시작 위치를 2중 포문으로 돌리면서, 시작 위치 가능한 부분에 대해서 전부 살펴보는 방식으로 진행했다. 또한, 이중 포문 안에 여러가지 조건문을 적게 된다면, 굉장히 코드가 난잡해지고 시간이 많이 걸릴 것 같아 W와 B값을 스왑해주는 방식으로 코드를 작성했다. 코드는 아래 참고

import sys
n,m = map(int,sys.stdin.readline().split())
my_map = []
for i in range(n) :
    my_map.append(str(sys.stdin.readline().rstrip()))

answer = n*m

for p in range(0,n-7) :
    for o in range(0,m-7) :
        r = p
        c = o
        w_str = 'W'
        w_cnt = 0
        b_str = 'B'
        b_cnt = 0

        for i in range(r,r+8) :
            for j in range(c,c+8) :
                if my_map[i][j] != w_str :
                    w_cnt +=1
                if my_map[i][j] != b_str :
                    b_cnt +=1
                temp = w_str
                w_str = b_str
                b_str = temp
            temp = w_str
            w_str = b_str
            b_str = temp
        if answer > b_cnt :
            answer = b_cnt
        if answer > w_cnt :
            answer = w_cnt


print(answer)