[백준] 3085번 - 사탕 게임 Python

문제

---

상근이는 어렸을 적에 "봄보니 (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

풀이

---

# N^3 풀이
def check(a,start_row, end_row, start_col, end_col):
    n = len(a)
    ans = 1
    for i in range(start_row, end_row+1):
        cnt = 1
        for j in range(1,n):
            if a[i][j] == a[i][j-1]:
                cnt+=1
            else:
                cnt = 1
            if ans<cnt:
                ans = cnt
    for i in range(start_col, end_col+1):
        cnt =1
        for j in range(1,n):
            if a[j][i] == a[j-1][i]:
                cnt+=1
            else:
                cnt =1
            if ans<cnt:
                ans = cnt
    return ans

n = int(input())
a = [list(input()) for _ in range(n)]
ans = 0

# 상하좌우로 바꿀 수 있는데 왼쪽이랑 위로 바꾸는 방법은 이미 이전단계에서 비교함 
# 따라서 오른쪽, 아래쪽만 바꿔서 비교해나간다
for i in range(n):
    for j in range(n):
        # 열 바꾸기
        if j+1<n:
            a[i][j],a[i][j+1] = a[i][j+1],a[i][j]
            tmp = check(a,i,i,j,j+1)
            if ans< tmp:
                ans= tmp
            a[i][j],a[i][j+1] = a[i][j+1],a[i][j]
        # 행 바꾸기
        if i+1<n:
            a[i][j], a[i+1][j] = a[i+1][j],a[i][j]
            tmp = check(a,i,i+1,j,j)
            if ans<tmp:
                ans = tmp;
            a[i][j], a[i+1][j] = a[i+1][j],a[i][j]
print(ans)

정리

---

• 오른쪽과 아래쪽만 바꾸고 비교한다.
• 바뀐 부분을 포함한 열과 행들만 확인하면 중복되는 부분을 줄일 수 있다.