[BOJ] 3085 : 사탕게임 (Python)

문제

상근이는 어렸을 적에 "봄보니 (Bomboni)" 게임을 즐겨했다.

가장 처음에 N×N크기에 사탕을 채워 놓는다. 사탕의 색은 모두 같지 않을 수도 있다. 상근이는 사탕의 색이 다른 인접한 두 칸을 고른다. 그 다음 고른 칸에 들어있는 사탕을 서로 교환한다. 이제, 모두 같은 색으로 이루어져 있는 가장 긴 연속 부분(행 또는 열)을 고른 다음 그 사탕을 모두 먹는다.

사탕이 채워진 상태가 주어졌을 때, 상근이가 먹을 수 있는 사탕의 최대 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 보드의 크기 N이 주어진다. (3 ≤ N ≤ 50)

다음 N개 줄에는 보드에 채워져 있는 사탕의 색상이 주어진다. 빨간색은 C, 파란색은 P, 초록색은 Z, 노란색은 Y로 주어진다.

사탕의 색이 다른 인접한 두 칸이 존재하는 입력만 주어진다.

출력

첫째 줄에 상근이가 먹을 수 있는 사탕의 최대 개수를 출력한다.

해결 방안

• 인접한 칸을 교환:
  • 각 칸에 대해 가로와 세로로 인접한 칸과 사탕 색이 다를 경우 교환
  • 교환한 후에 연속된 같은 색의 사탕 개수를 확인
• 원래 상태로 복구:
  • 사탕을 교환하여 개수를 센 후에는 다시 원래 상태로 복구
  • 이 과정을 반복하여 모든 칸에서 최대 사탕 개수를 확인
• 최대 개수 업데이트:
  • 교환할 때마다 얻는 최대 연속 개수를 비교하여 최댓값을 기록

N = int(input())

board = [list(input()) for _ in range(N)]

result = 0

def max_check():
    global result
    for i in range(N):
        cnt = 1
        for j in range(N-1):
            if board[i][j] == board[i][j+1]:
                cnt += 1
                result = max(result, cnt)
            else:
                cnt = 1

    for j in range(N):
        cnt = 1
        for i in range(N-1):
            if board[i][j] == board[i+1][j]:
                cnt += 1
                result = max(result, cnt)
            else:
                cnt = 1


for i in range(N):
    for j in range(N-1):
        board[i][j], board[i][j+1] = board[i][j+1], board[i][j]
        max_check()
        board[i][j+1], board[i][j] = board[i][j], board[i][j+1]

        board[j][i], board[j+1][i] = board[j+1][i], board[j][i]
        max_check()
        board[j][i], board[j+1][i] = board[j+1][i], board[j][i]
print(result)