문제
지민이는 자신의 저택에서 MN개의 단위 정사각형으로 나누어져 있는 M×N 크기의 보드를 찾았다. 어떤 정사각형은 검은색으로 칠해져 있고, 나머지는 흰색으로 칠해져 있다. 지민이는 이 보드를 잘라서 8×8 크기의 체스판으로 만들려고 한다.
체스판은 검은색과 흰색이 번갈아서 칠해져 있어야 한다. 구체적으로, 각 칸이 검은색과 흰색 중 하나로 색칠되어 있고, 변을 공유하는 두 개의 사각형은 다른 색으로 칠해져 있어야 한다. 따라서 이 정의를 따르면 체스판을 색칠하는 경우는 두 가지뿐이다. 하나는 맨 왼쪽 위 칸이 흰색인 경우, 하나는 검은색인 경우이다.
보드가 체스판처럼 칠해져 있다는 보장이 없어서, 지민이는 8×8 크기의 체스판으로 잘라낸 후에 몇 개의 정사각형을 다시 칠해야겠다고 생각했다. 당연히 8*8 크기는 아무데서나 골라도 된다. 지민이가 다시 칠해야 하는 정사각형의 최소 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 N과 M이 주어진다. N과 M은 8보다 크거나 같고, 50보다 작거나 같은 자연수이다. 둘째 줄부터 N개의 줄에는 보드의 각 행의 상태가 주어진다. B는 검은색이며, W는 흰색이다.
출력
첫째 줄에 지민이가 다시 칠해야 하는 정사각형 개수의 최솟값을 출력한다.
내 제출
row, col = map(int, input().split(" "))
lst = []
count = []
for i in range(row):
lst.append(input())
for a in range(row-7):
for b in range(col-7):
count1 = 0
count2 = 0
for i in range(a, a+8) : # 시작점이 W인 경우
for j in range(b, b+8) :
if i % 2 == 0 :
if j % 2 == 0 :
if lst[i][j] != 'W' :
count1 += 1
else :
if lst[i][j] != 'B' :
count1 += 1
else :
if j % 2 == 0 :
if lst[i][j] != 'B' :
count1 += 1
else :
if lst[i][j] != 'W' :
count1 += 1
for i in range(a, a+8) : # 시작점이 B인 경우
for j in range(b, b+8) :
if i % 2 == 0 :
if j % 2 == 0 :
if lst[i][j] != 'B' :
count2 += 1
else :
if lst[i][j] != 'W' :
count2 += 1
else :
if j % 2 == 0 :
if lst[i][j] != 'W' :
count2 += 1
else :
if lst[i][j] != 'B' :
count2 += 1
count.append(min(count1, count2))
print(min(count))
해설
row, col = map(int, input().split(" "))
lst = []
count = []
for i in range(row):
lst.append(input())입력받을 체스판의 가로와 세로를 입력받는다.
그리고 한 줄마다 lst에 입력해 2차원 리스트로 입력받는다.
for a in range(row-7):
for b in range(col-7):
count1 = 0
count2 = 0
for i in range(a, a+8) : # 시작점이 W인 경우
for j in range(b, b+8) :
if i % 2 == 0 :
if j % 2 == 0 :
if lst[i][j] != 'W' :
count1 += 1
else :
if lst[i][j] != 'B' :
count1 += 1
else :
if j % 2 == 0 :
if lst[i][j] != 'B' :
count1 += 1
else :
if lst[i][j] != 'W' :
count1 += 1반복문을 통해 a와 b를 각각 0~row-8과 0~col-8만큼 반복한다.
이는 체스판을 8*8만큼 자를 수 있는 체스판의 (0, 0)원소를 구하는 것이다.
count1과 count2를 0으로 초기화시킨다.
count1은 W로 시작하는 경우 추가적으로 칠해야 하는 수이고, count2는 B로 시작하는 경우 추가적으로 칠해야 하는 수이다.
위에서 구한 (a, b)를 체스판의 (0, 0)으로 대입하여 반복문을 통해 추가적으로 칠해야 하는 수를 구한다.
위의 경우는 W로 시작하는 경우이다.
체스판을 칠하는 경우
W B
B W
모양이 반복되어야 하기 때문에 해당 경우가 아닐 때 count1에 +1한다.
for i in range(a, a+8) : # 시작점이 B인 경우
for j in range(b, b+8) :
if i % 2 == 0 :
if j % 2 == 0 :
if lst[i][j] != 'B' :
count2 += 1
else :
if lst[i][j] != 'W' :
count2 += 1
else :
if j % 2 == 0 :
if lst[i][j] != 'W' :
count2 += 1
else :
if lst[i][j] != 'B' :
count2 += 1위에서 구한 (a, b)를 체스판의 (0, 0)으로 대입하여 반복문을 통해 추가적으로 칠해야 하는 수를 구한다.
위의 경우는 B로 시작하는 경우이다.
체스판을 칠하는 경우
B W
W B
모양이 반복되어야 하기 때문에 해당 경우가 아닐 때 count2에 +1한다.
count.append(min(count1, count2))
print(min(count))W로 시작하는 경우와 B로 시작하는 경우 중 작은 것을 count 리스트에 추가한다.
이렇게 추가하면 8*8로 자를 수 있는 경우의 개수만큼 count 리스트에 원소가 추가되었다.
해당 경우 중 가장 작은 것을 출력한다.
p.s. W와 B로 나누는 경우를 (i+j) % 2 == 0 과 != 0인 경우로 나눠서 축약하여 생각할 수 있다.
차후 시간이 된다면 코드를 개량해 보는 것도 좋을 것 같다.