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

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

문제

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

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

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

입력

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

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

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

출력

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

n=int(input())
data=[list(input()) for _ in range(n)]
answer=0

# 테이블 전부 체크해주는 함수
def count(arr):
  n=len(arr)
  answer=1

  for i in range(n):
    # 가로줄 순회, 연속되는 수 세기
    cnt=1
    for j in range(1,n):
    # 이전과 같다면 +1
      if arr[i][j]==arr[i][j-1]:
        cnt+=1
     #다르면 다시 1로 초기화
      else:
        cnt=1
	#현재 cnt가 더 크면 갱신
      if cnt>answer:
        answer=cnt
     
    #세로줄 순회, 연속 되는 수 세기
    cnt=1
    for j in range(1,n):
      if arr[j][i]==arr[j-1][i]:
        cnt+=1
      else:
        cnt=1
      if cnt>answer:
        answer=cnt

  return answer

for i in range(n):
  for j in range(n):
  #가로줄 바꾸기
    if j+1<n:
    #인접한 것과 바꿈
      data[i][j], data[i][j+1]=data[i][j+1], data[i][j]
      #바꿨을때 가장 긴 연속 부분 체크
      tmp=count(data)
      if tmp>answer:
        answer=tmp
      #바꿨던것 다시 원래대로
      data[i][j], data[i][j+1]=data[i][j+1], data[i][j]
      
	#세로줄 바꾸기
    if i+1<n:
      data[i][j], data[i+1][j]=data[i+1][j], data[i][j]
      tmp=count(data)
      if tmp>answer: 
        answer=tmp

      data[i][j], data[i+1][j]=data[i+1][j], data[i][j]

print(answer)

이게 브루트 포스구나...!!