[BOJ] #3085_사탕 게임(Python3, 파이썬)

본 게시글은 백준 저지 알고리즘 문제를 '파이썬, Python3' 언어로 풀이한 내용을 주관적으로 정리하였으며, 내용과 관련된 코드리뷰 및 피드백 환영합니다.

티어

Silver III

문제

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

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

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

입력

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

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

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

출력

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

예제 입력 및 출력 1

3
CCP
CCP
PPC

3

예제 입력 및 출력 2

4
PPPP
CYZY
CCPY
PPCC

4

예제 입력 및 출력 3

5
YCPZY
CYZZP
CCPPP
YCYZC
CPPZZ

4

---

풀이

import sys 
input = sys.stdin.readline

def count():
    global maxi
    for i in range(n):
        cnt = 1
        for j in range(1, n):
            if data[i][j] == data[i][j-1]:
                cnt += 1
                maxi = max(maxi, cnt)
            else:
                cnt = 1
        cnt = 1
        for j in range(1, n):
            if data[j][i] == data[j-1][i]:
                cnt += 1
                maxi = max(maxi, cnt)
            else:
                cnt = 1

n = int(input()) # 개수 
data = [list(input()) for _ in range(n)]
maxi = 0

for i in range(n):
    for j in range(n):  
        # 열
        if j+1 < n:
            # 인접한 것 바꾸기
            data[i][j], data[i][j+1] = data[i][j+1], data[i][j]
            count()
            data[i][j], data[i][j+1] = data[i][j+1], data[i][j]

        # 행
        if i+1 < n:
            data[i][j], data[i+1][j] = data[i+1][j], data[i][j]
            count()
            data[i][j], data[i+1][j] = data[i+1][j], data[i][j]

print(maxi)

처음 문제를 잘못 이해해서 접근하는 데 시간이 좀 걸렸다. 이 문제는 브루트포스 알고리즘에 딱 알맞는 문제처럼 보였다. 하나 하나 다 탐색하고 확인하고, 다시 되돌려두는 꼼꼼한 절차를 밟아야했기 때문이다.... 크게 행과 열로 생각해서 풀었다. 데이터를 좌우, 상하로 바꾸어보면서 count함수로 올리고, 그 중 가장 긴 길이를 가지는 값을 maxi 값으로 바꿔주면 된다. 

* 시간 초과가 뜬 이유는 함수 밖과 안에서 한 번씩 더 max함수를 사용하려 했기 때문인 것 같다.