[백준] 1018 : 체스판 다시 칠하기 (Python/파이썬)

문제 링크

https://www.acmicpc.net/problem/1018

문제

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

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

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

입력

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

출력

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

예제

입력

8 8
WBWBWBWB
BWBWBWBW
WBWBWBWB
BWBBBWBW
WBWBWBWB
BWBWBWBW
WBWBWBWB
BWBWBWBW

출력

1

입력

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

출력

12

---

정답 및 풀이

import sys

input = sys.stdin.readline

n, m = map(int, input().split())
board = [input().strip() for _ in range(n)]

def count_repaint(x, y):
    case1 = 0  # W로 시작하는 경우
    case2 = 0  # B로 시작하는 경우

    for i in range(8):
        for j in range(8):
            current = board[x + i][y + j]

            # (i+j) 짝수면 시작 색, 홀수면 반대 색
            # W 시작
            if (i + j) % 2 == 0:
                if current != 'W':
                    case1 += 1
            else:
                if current != 'B':
                    case1 += 1

            # B 시작
            if (i + j) % 2 == 0:
                if current != 'B':
                    case2 += 1
            else:
                if current != 'W':
                    case2 += 1

    return min(case1, case2)

answer = float('inf')

for i in range(n - 8 + 1):
    for j in range(m - 8 + 1):
        answer = min(answer, count_repaint(i, j))

print(answer)

1. import sys
   • 표준 입력을 더 빠르게 처리하기 위해 sys 모듈을 가져온다.
2. input = sys.stdin.readline
   • 기본 input()보다 빠른 입력을 사용하도록 설정한다.
3. n, m = map(int, input().split())
   • 체스판의 세로 크기 n, 가로 크기 m을 정수로 입력받는다.
4. board = [input().strip() for _ in range(n)]
   • 체스판의 각 줄을 읽어 리스트로 저장한다.
   • strip()으로 줄 끝의 개행문자를 제거한다.
5. def count_repaint(x, y):
   • (x, y)를 시작점으로 하는 8×8 부분 보드를 다시 칠해야 하는 횟수를 계산하는 함수다.
6. case1 = 0 / case2 = 0
   • W로 시작하는 체스판, B로 시작하는 체스판 각각에서 다시 칠해야 하는 칸 개수를 저장한다.
7. for i in range(8):
   • 8×8 부분의 세로 방향 칸들을 탐색한다.
8. for j in range(8):
   • 8×8 부분의 가로 방향 칸들을 탐색한다.
9. current = board[x + i][y + j]
   • 실제 체스판에서 (x+i, y+j) 위치의 문자를 가져온다.
10. (i + j) % 2 == 0
    • 8×8 내부 좌표 기준으로 짝수 칸인지 판단한다.
    • 체스판은 짝수 칸과 홀수 칸이 서로 다른 색이므로 이를 이용한다.
11. W로 시작하는 경우 검사 (case1)
    • 짝수 칸은 W여야 한다.
    • 홀수 칸은 B여야 한다.
12. if (i + j) % 2 == 0:
    • 짝수 칸이면
      • if current != 'W': case1 += 1
      • 현재 색이 W가 아니면 다시 칠해야 하므로 +1 한다.
13. else:
    • 홀수 칸이면
      • if current != 'B': case1 += 1
      • 현재 색이 B가 아니면 다시 칠해야 하므로 +1 한다.
14. B로 시작하는 경우 검사 (case2)
    • 짝수 칸은 B여야 한다.
    • 홀수 칸은 W여야 한다.
15. 짝수 칸에서 if current != 'B': case2 += 1
    • B가 아닐 경우 다시 칠해야 하므로 +1 한다.
16. 홀수 칸에서 if current != 'W': case2 += 1
    • W가 아닐 경우 다시 칠해야 하므로 +1 한다.
17. return min(case1, case2)
    • 두 경우 중 더 적은 칠하기 횟수를 반환한다.
18. answer = float('inf')
    • 최소값 비교를 위해 answer를 매우 큰 값(무한대)으로 초기화한다.
19. for i in range(n - 8 + 1):
    • 8×8로 잘라낼 수 있는 세로 방향 시작점을 모두 탐색한다.
20. for j in range(m - 8 + 1):
    • 8×8로 잘라낼 수 있는 가로 방향 시작점을 모두 탐색한다.
21. answer = min(answer, count_repaint(i, j))
    • 현재 (i, j) 8×8 구간에서 필요한 칠하기 횟수와 기존 최소값을 비교해 더 작은 값을 저장한다.
22. print(answer)
    • 모든 구간을 검사한 후, 가장 작은 칠하기 횟수를 출력한다.

---

새롭게 배운 내용 및 느낀점

• float(’inf’)
  • 파이썬에서 양의 무한대를 뜻한다.
  • 어떤 값과 비교해도 항상 더 크기 때문에 사용한다.