백준 3085 : 사탕 게임 (Python)

백준 3085 문제

문제

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

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

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

입력

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

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

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

출력

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

예제 입력 1

3
CCP
CCP
PPC

예제 출력 1

3

---

나의 풀이

• 구현을 오랜만에 풀이하려니 다른 알고리즘보다 더 복잡하게 느껴지는 것 같다. 처음에는 행 또는 열별로 접근해 실제로 2차원 배열을 건드리지 않고 풀이하려고 했는데 오히려 이 방식이 더 복잡하게 느껴졌다. 따라서 완전탐색으로 사탕 위치를 일일이 바꾸어가며 배열을 검사하는 방식으로 구현했다. for문이 4번 중복되어 시간복잡도가 높을 것을 걱정할 수 있으나 시간 제한 1초에 N이 최대 50이기에 625*(10^4) < 10^8 을 만족해 시간 초과가 발생하지 않는다. 파이썬으로 구현해 swab을 매우 간편하게 처리할 수 있었다.

import sys 
from collections import Counter 

answer = 0

def check(N,colors):
    max_num = 0 
    for i in range(N):
        cnt = 1 
        for j in range(1,N):
            if colors[i][j] == colors[i][j-1]:
                cnt +=1 
                max_num = max(max_num,cnt)
            else:
                cnt = 1
        
    for j in range(N):
        cnt = 1 
        for i in range(1,N):
            if colors[i][j] == colors[i-1][j]:
                cnt +=1 
                max_num = max(max_num,cnt)
            else:
                cnt = 1
        
    return max_num
        
                

N = int(sys.stdin.readline())
colors =[list(map(str,sys.stdin.readline().strip())) for _ in range(N)]

for i in range(N):
    for j in range(N-1):
        if colors[i][j]!=colors[i][j+1]:
            colors[i][j], colors[i][j+1] = colors[i][j+1], colors[i][j]
            answer = max(answer,check(N,colors))
            colors[i][j], colors[i][j+1] = colors[i][j+1], colors[i][j]

for i in range(N-1):
    for j in range(N):
        if colors[i][j]!=colors[i+1][j]:
            colors[i][j], colors[i+1][j] = colors[i+1][j], colors[i][j]
            answer = max(answer,check(N,colors))        
            colors[i][j], colors[i+1][j] = colors[i+1][j], colors[i][j]

print(answer)