[백준] 3085번 사탕 게임- 파이썬(Python)

문제

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

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

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

입력

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

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

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

 

출력

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

문제 링크: https://www.acmicpc.net/problem/3085

3085번: 사탕 게임

 

    import sys
    input = sys.stdin.readline

    n = int(input())
    arr = [list(input().rstrip()) for _ in range(n)]
    answer = 0
    def check(arr):
        answer = 1
        for i in range(n):
            count = 1
            for j in range(1, n):
                if arr[i][j] == arr[i][j-1]:
                    count += 1
                else:
                    count = 1
                if count > answer:
                    answer = count
            count = 1
            for j in range(1, n):
                if arr[j-1][i] == arr[j][i]:
                    count += 1
                else:
                    count = 1
                if count > answer:
                    answer = count
        return answer

    for i in range(n):
        for j in range(n):
            if j+1 < n:
                arr[i][j], arr[i][j+1] = arr[i][j+1], arr[i][j]
                answer = max(answer, check(arr))
                arr[i][j], arr[i][j+1] = arr[i][j+1], arr[i][j]
            if i+1 < n:
                arr[i][j], arr[i+1][j] = arr[i+1][j], arr[i][j]
                answer = max(answer, check(arr))
                arr[i][j], arr[i+1][j] = arr[i+1][j], arr[i][j]

    print(answer)

사탕을 교환하는 방법은 4가지이지만, 위와 아래를 바꾸는 것이나 아래와 위를 바꾸는 것은 똑같고, 좌+우 교환도 우+좌와 같기때문에 2가지 방향만 확인하면 된다.

연속되는 문자를 계속 세서 count에 저장해주고, 연속되지 않는 문자가 발견될시 즉시 count를 1로 리셋시키면 된다

 

그리고 n x n인 arr를 하나하나 1. 바꿔주고, 2. 가장 긴 열/행을 체크하고 3. 원래 상태로 돌려놓는다.