[Python] 백준 / 사탕 게임 / 3085번 / 브루트포스

문제

사탕 게임 문제 링크

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

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

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

입력

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

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

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

5
YCPZY
CYZZP
CCPPP
YCYZC
CPPZZ

출력

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

4

• 접근 방식

1. 2차원 배열로 사탕을 입력받는다

2. 이중 반복문으로 각 원소에 접근한 뒤 행 또는 열 방향으로 뒤의 사탕과 자리를 바꿔준다

3. 자리를 바꾼 뒤 사탕 배열의 행, 열 최대 사탕 수를 구한다

4. 다시 자리를 바꿔준다

코드

# url : https://www.acmicpc.net/problem/3085
# 난이도 : silver 4 ????

import sys

n = int(input())

candy = [[*sys.stdin.readline().strip()] for _ in range(n)]

answer = 0

def h():
    global answer, n, candy

    for i in range(n):
        sub = 1
        for j in range(n-1):
            if candy[j][i] == candy[j+1][i] :
                sub += 1
            else:
                if answer < sub :
                    answer = sub
                sub = 1

        if answer < sub:
            answer = sub

def w():
    global answer, n, candy

    for i in range(n):
        sub = 1
        for j in range(n-1):
            if candy[i][j] == candy[i][j+1]:
                sub += 1
            else :
                if answer < sub :
                    answer = sub
                sub = 1
        
        if answer < sub:
            answer = sub

for i in range(n):
    for j in range(n):
        if j < n-1 :
            candy[i][j], candy[i][j+1] = candy[i][j+1], candy[i][j]
            h()
            w()
            candy[i][j], candy[i][j+1] = candy[i][j+1], candy[i][j]
        if i < n-1:
            candy[i][j], candy[i+1][j] = candy[i+1][j], candy[i][j]
            h()
            w()
            candy[i][j], candy[i+1][j] = candy[i+1][j], candy[i][j]


print(answer)