[백준] 3085 : 사탕게임 - Python

사탕게임

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

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

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

입력

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

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

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

출력

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

import sys
sys.stdin = open("input.text", "rt")

n = int(input())
data = [list(input()) for _ in range(n)]

#가장 중요한 것은 모든 경우의 수 확인
# 1. 먼저 인접한 사탕 교환 (상하, 좌우) 그리고
# 2. 개수 세기

def change():
    max_cnt = 0
    for i in range(n):
        cnt =1 
        for j in range(n-1):
            if data[i][j] == data[i][j+1]:
                cnt += 1
            else:
                max_cnt = max(max_cnt, cnt)
                cnt = 1
        max_cnt = max(max_cnt, cnt)

        cnt = 1
        for j in range(n-1):
            if data[j][i] == data[j+1][i]:
                cnt += 1
            else:
                max_cnt = max(max_cnt, cnt)
                cnt = 1
        max_cnt = max(max_cnt, cnt)
    return max_cnt

res = 0
for i in range(n):
    for j in range(n):
        if j < n-1:
            data[i][j], data[i][j+1] = data[i][j+1], data[i][j]
            t = change()
            res = max(res, t)
            data[i][j], data[i][j+1] = data[i][j+1], data[i][j]
        if i < n-1:
            data[i][j], data[i+1][j] = data[i+1][j], data[i][j]
            t = change()
            res = max(res, t)
            data[i][j], data[i+1][j] = data[i+1][j], data[i][j]

print(res)

😀 코멘트

해당 문제를 풀 때 굉장히 당황했다. 문제를 풀 때 n이 50인 것을 보고 완전탐색으로 전부 확인하면 된다고 생각했다. 근데 중요한게 문제가 너무 안되는 것을 보고.. 무엇이 문제일까 고민하다가 다른 사람의 답을 보고 내 문제를 확인했다. 계속해서 max_cnt를 갱신해줘야 했다...