[백준] / [Python] / [3085] 사탕 게임

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

---

문제

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

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

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

입력

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

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

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

출력

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

---

완전 탐색

import sys
input = sys.stdin.readline

# 가장 긴 연속 부분
def longest_increasing_subsequence(arr: list):
    max_ = 0
    
    # 행
    for y in range(N):
        cnt = 1
        prev = arr[y][0]
        
        for next in arr[y][1:]:
            if prev == next:
                cnt += 1
            else:
                prev = next
                cnt = 1            
            max_ = max(max_, cnt)
    
    # 열
    temp = list(map(list, zip(*arr)))
    
    for y in range(N):
        cnt = 1
        prev = temp[y][0]
        
        for next in temp[y][1:]:
            if prev == next:
                cnt += 1
            else:
                prev = next
                cnt = 1            
            max_ = max(max_, cnt)
        
    return max_

N = int(input())

arr = [list(input().strip()) for _ in range(N)]
visited = [[False for _ in range(N)] for _ in range(N)]

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

max_ = 0

for y in range(N):
    for x in range(N):
        if visited[y][x]: continue
        visited[y][x] = True
        
        for i in range(4):
            nx, ny = x + dx[i], y + dy[i]
            
            if -1 < nx < N and -1 < ny < N and visited[ny][nx] == False:
                if arr[y][x] != arr[ny][nx]:
                    arr[y][x] , arr[ny][nx] = arr[ny][nx], arr[y][x]
                    
                    value = longest_increasing_subsequence(arr)
                    max_ = max(max_, value)
                    
                    arr[y][x] , arr[ny][nx] = arr[ny][nx], arr[y][x]

print(max_)