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

문제:

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

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

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

 

입력:

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

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

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

 

출력:

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

 

👇간단  풀이 설명

1. 사탕을 2차원 리스트로 받는다.

2. 2중 for 문을 돌면서 가로 양옆의 사탕을 바꾸고 몇개 먹을 수 있는지 카운트 한다. 세로 위아래의 사탕을 바꾸고 몇개 먹을 수 있는지 카운트 한다. 바꾼 로직 안에서 정의한 카운트 함수를 사용한다.

3. 2중 for문 안에서 모든 사탕을 바꾸며 카운트 함수를 돌렸으면 카운트 함수 내에서 가장 많이 먹을 수 있는 사탕 수에 대해서는 계속 갱신이 되었기 때문에, 해당 값을 출력 하면 된다.

 

👇내 풀이(정답 여부 O) 

n = int(input())
candies=[]
ans = 1
 
for i in range(n):
    temp =[]
    temp_str = input()
    for j in range(n):
        temp.append(temp_str[j])
    candies.append(temp)
    

    
# 몇개 먹을 수 있는지 찾는 함수
def search():
    global ans
    for i in range(n):
        cnt = 1
        for j in range(n-1):
            if candies[i][j]== candies[i][j+1]:
                cnt+=1
                ans = max(cnt,ans)
            else:
                cnt = 1
        #ans = max(cnt,ans)
        
    for i in range(n):
        cnt = 1
        for j in range(n-1):
            if candies[j][i] == candies[j+1][i]:
                cnt+=1
                ans = max(cnt,ans)
            else:
                cnt = 1
        #ans = max(cnt,ans)
        


# [모든 인접한 두 자리 뒤집어보고 찾기]    
# 가로 뒤집기
for i in range(n):
    for j in range(n-1):
        candies[i][j],candies[i][j+1] = candies[i][j+1],candies[i][j]
        search()
        candies[i][j],candies[i][j+1] = candies[i][j+1],candies[i][j]

# 세로 뒤집기
for i in range(n):
    for j in range(n-1):
        candies[j][i],candies[j+1][i] = candies[j+1][i],candies[j][i]
        search()
        candies[j][i],candies[j+1][i] = candies[j+1][i],candies[j][i]

        
print(ans)