백준 3085번: 사탕 게임 (실버 2)

https://www.acmicpc.net/problem/3085

알고리즘 분류

• 구현
• 브루트포스 알고리즘

제출 코드

def search_max_candy(board, n):
    max_candy = 1
    # 행 방향 점검
    for i in range(n):
        row_candy = 1
        for j in range(n-1):
            if board[i][j] == board[i][j+1]:
                row_candy += 1
            else:
                row_candy = 1
            max_candy = max(max_candy, row_candy)
        if max_candy == n:
            return max_candy
            
    # 열 방향 점검
    for j in range(n):
        col_candy = 1
        for i in range(n-1):
            if board[i][j] == board[i+1][j]:
                col_candy += 1
            else:
                col_candy = 1
            max_candy = max(max_candy, col_candy)
        if max_candy == n:
            return max_candy
        
    return max_candy

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

for _ in range(n):
    input_str = list(input())
    board.append(input_str)

max_candy = 1

# 교환할 사탕 칸 선택    
for i in range(n):
    for j in range(n):
        b_candy, r_candy = 1, 1  # 각 교환마다 초기화

        # 아래에 있는 사탕과 교환하는 경우
        if i != n-1 and board[i][j] != board[i+1][j]:
            board[i][j], board[i+1][j] = board[i+1][j], board[i][j]
            b_candy = search_max_candy(board, n)
            # 보드 원상 복귀
            board[i][j], board[i+1][j] = board[i+1][j], board[i][j]            

        # 우측에 있는 사탕과 교환하는 경우
        if j != n-1 and board[i][j] != board[i][j+1]:
            board[i][j], board[i][j+1] = board[i][j+1], board[i][j]
            r_candy = search_max_candy(board, n)
            # 보드 원상 복귀
            board[i][j], board[i][j+1] = board[i][j+1], board[i][j]

        max_candy = max(max_candy, b_candy, r_candy)
        
        # 사탕의 최대 연속 개수가 n이면 바로 종료
        if max_candy == n:
            break

    if max_candy == n:
        break

print(max_candy)

 

문제

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

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

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

입력

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

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

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

출력

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

예제 입력 1  

3
CCP
CCP
PPC

예제 출력 1  

3

예제 입력 2  

4
PPPP
CYZY
CCPY
PPCC

예제 출력 2  

4

예제 입력 3  

5
YCPZY
CYZZP
CCPPP
YCYZC
CPPZZ

예제 출력 3  

4