[Baekjoon] 3085번 사탕 게임

문제

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

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

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

입력

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

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

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

출력

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

예제

Example 1:

Input: 
5
YCPZY
CYZZP
CCPPP
YCYZC
CPPZZ
Output: 
4

조건

시간 제한 : 1초

메모리 제한 : 128 MB

풀이과정

풀이 1

구현이 힘들었던 문제이다. 2중 for문을 활용하며, 각 좌표별로 오른쪽, 아래 좌표를 비교한다. (범위를 벗어나는 경우 제외) 서로 다른 경우에는 서로 값을 교환해준 뒤 cal_max함수를 통해, 값을 구한다. 값을 구하면 서로 값을 다시 돌려준다.

한편, cal_max함수는 해당 좌표에서 가로 세로 줄을 구해서 사탕이 모두 같은 색으로 이루어져 있는 가장 긴 연속 부분의 최대를 구해주는 함수이다.

n = int(input())
graph = []

for _ in range(n):
    graph.append(list(input()))

ans = 1
dx = [0, 0, 1, -1]
dy = [1, -1, 0, 0]

def cal_max(x, y):
    ans = 1
    arr = graph[x]
    cnt1 = 1
    for i in range(n-1):
        if arr[i] == arr[i+1]:
            cnt1 += 1
        else:
            cnt1 = 1
        ans = max(ans, cnt1)

    arr = []
    for i in range(n):
        arr.append(graph[i][y])

    cnt2 = 1
    for i in range(n-1):
        if arr[i] == arr[i+1]:
            cnt2 += 1
        else:
            cnt2 = 1
        ans = max(ans, cnt2)

    return ans
for x in range(n):
    for y in range(n):
        # 위아래로 비교
        nx, ny = x+1, y
        if 0 <= nx < n and 0 <= ny < n:
            graph[x][y], graph[nx][ny] = graph[nx][ny], graph[x][y]
            ans = max(ans, cal_max(x, y))
            ans = max(ans, cal_max(nx, ny))
            graph[x][y], graph[nx][ny] = graph[nx][ny], graph[x][y]
        # 옆으로 비교
        nx, ny = x, y+1
        if 0 <= nx < n and 0 <= ny < n:

            graph[x][y], graph[nx][ny] = graph[nx][ny], graph[x][y]

            ans = max(ans, cal_max(x, y))
            ans = max(ans, cal_max(nx, ny))
            graph[x][y], graph[nx][ny] = graph[nx][ny], graph[x][y]

print(ans)