출처: https://www.acmicpc.net/problem/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())
board = []
for i in range(n):
board.append(input())
wMin = bMin = tempMin = 0 #W의 경우 최소값과 B의 경우 최소값, 임시 최소값
min = 64
for i in range(n-7):
for j in range(m-7):
for k in range(i, i+8):
# W의 경우
for l in range(j, j+8):
if (k+l-i-j) % 2 == 0: # 이때는 보드판이 W여야 함
if board[k][l] != 'W':
wMin += 1
else:
if board[k][l] != 'B':
wMin += 1
# B의 경우
for l in range(j, j+8):
if (k+l-i-j) % 2 == 0: # 이때는 보드판이 B여야 함
if board[k][l] != 'B':
bMin += 1
else:
if board[k][l] != 'W':
bMin += 1
tempMin = wMin if wMin < bMin else bMin
min = tempMin if tempMin < min else min
wMin = bMin = tempMin = 0
print(min)
풀이
전체 개요
- 입력받은 배열에서 우선 맨 위 왼쪽 8*8 개의 최소값을 구한다. 구한 최소값을 min이라는 변수에 저장한다.
- 구하는 범위를 한 칸 오른쪽으로 옮겨 해당 범위의 최소값을 구한다. 이때 최소값이 이 전에 구한 최소값보다 작을 경우 min 값을 바꾼다.
- m-7 범위까지 모두 계산해 더이상 오른쪽에 남은 범위가 없으면 구하는 범위를 한 칸 밑으로 옮긴다.
- n-7 범위까지 구한 후 최종적인 최소값을 구한다.
최소값 구하기
문제에 나왔듯 맨 왼쪽 위쪽 칸이 흰색(W)인 경우와, 검은색(B)인 경우로 나누어 최소값을 구해야 한다.
- W의 경우
- (k+l-i-j) % 2 == 0 일 때, 즉 해당 칸의 k+l의 값이 짝수일 때는 W가 와야 한다. (i와 j는 칸을 옮겼을 경우 더하기 때문에 다시 빼줘야 함)
- board[k][l] != ‘W’이면 w의 경우의 최소값인 wMin += 1을 한다.
- k+l 값이 홀수일 때도 != ‘B’이면 wMin += 1을 한다.
- B의 경우
- 마찬가지. 칸에 B 혹은 W가 오지 않으면 bMin += 1을 한다.