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

문제

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

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

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

입력

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

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

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

출력

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

Approach

• 개빡센 구현 문제였다.. 싫다 ㅜ
• check라는 함수를 사용한다 -> 현재 상태에서 가장 긴 행이나 열의 길이를 반환해주는 함수
• 오른쪽과 swap하는 경우 + 아래쪽과 swap하는 경우 다 검사하여 최댓값을 찾는다

Source Code

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int check(vector<string> & vec)
{
    int ans = 1;
    int cnt;
    
    for (int i = 0; i < vec.size(); i++)
    {
        cnt = 1;
        for (int j = 1; j < vec.size(); j++) // 긴 행 찾기 
        {
            if (vec[i][j] == vec[i][j - 1])
                cnt++;
            else
                cnt = 1;
            if (cnt > ans)  ans = cnt; // 최댓값 갱신 
        }
        
        cnt = 1;
        for (int j = 1; j < vec.size(); j++) // 긴 열 찾기 
        {
            if (vec[j][i] == vec[j - 1][i])
                cnt++;
            else
                cnt = 1;
            if (cnt > ans)  ans = cnt; // 최댓값 갱신 
        }
    }
    
    return ans;
}



int main() {
    
    int n;
    cin >> n;
    
    
    vector<string> v;
    for (int i = 0; i < n; i++)
    {
        string input;
        cin >> input;
        v.push_back(input);
    }
    
    
    int ans = 1;
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < n; j++)
        {
            if (j + 1 < n) // 사탕판 안이면 
            {
            	// 오른쪽과 교환 후 그 상태에서 가장 긴 행/열 길이 찾기 
                swap(v[i][j], v[i][j + 1]);
                int temp = check(v);
                if (temp > ans) ans = temp; // 최댓값 갱신
                swap(v[i][j], v[i][j + 1]); // 다시 원상복구 
            }
            
            if (i + 1 < n) // 사탕판 안이면 
            {
            	// 아래쪽과 교환 후 그 상태에서 가장 긴 행/열 길이 찾기 
                swap(v[i][j], v[i + 1][j]);
                int temp = check(v);
                if (temp > ans) ans = temp; // 최댓값 갱신
                swap(v[i][j], v[i + 1][j]); // 다시 원상복구 
            }
        }
    }
    
    cout << ans << '\n';
}