[백준 3085 - C++] 사탕 게임 : 브루트포스

https://www.acmicpc.net/problem/3085

3085번: 사탕 게임

문제

상근이는 어렸을 적에 "봄보니 (Bomboni)" 게임을 즐겨했다.
가장 처음에 N×N크기에 사탕을 채워 놓는다. 사탕의 색은 모두 같지 않을 수도 있다. 상근이는 사탕의 색이 다른 인접한 두 칸을 고른다. 그 다음 고른 칸에 들어있는 사탕을 서로 교환한다. 이제, 모두 같은 색으로 이루어져 있는 가장 긴 연속 부분(행 또는 열)을 고른 다음 그 사탕을 모두 먹는다.
사탕이 채워진 상태가 주어졌을 때, 상근이가 먹을 수 있는 사탕의 최대 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 보드의 크기 N이 주어진다. (3 ≤ N ≤ 50)
다음 N개 줄에는 보드에 채워져 있는 사탕의 색상이 주어진다. 빨간색은 C, 파란색은 P, 초록색은 Z, 노란색은 Y로 주어진다.
사탕의 색이 다른 인접한 두 칸이 존재하는 입력만 주어진다.

출력

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

풀이

아래나 오른쪽에 다른게 있는 경우 swap해주고 find_candy함수를 호출하여 그 상태에서의 연속된 사탕의 수의 최대를 구한다.
그런 뒤 나머지 맵을 탐색해야 하므로 다시 swap하여 원래 상태로 만들어 주어야 한다.

#include <iostream>
#include <cstdio>

char map[51][51];

int X[2] = {1, 0};
int Y[2] = {0, 1};

int find_candy(int n){
    int cnt= 1;
    int max_cnt = 1;
    
    for(int i=0; i<n; i++){
        cnt = 1;
        for(int j=0; j<n-1; j++){
            if(map[i][j] == map[i][j+1])
                cnt++;
            else
                cnt = 1;
            
            if(cnt > max_cnt)
                max_cnt = cnt;
        }
        
    }
    
    for(int j=0; j<n; j++){
        cnt = 1;
        for(int i=0; i<n-1; i++){
            if(map[i][j] == map[i+1][j])
                cnt++;
            else
                cnt = 1;
            
            if(cnt > max_cnt)
                max_cnt = cnt;
        }
        
    }
    
    return max_cnt;
}

int main (void) {
    int n;
    int c_x, c_y;
    int max_candy = 0;
    
    scanf("%d", &n);
    
    for(int i=0; i<n; i++){
        scanf("%s", map[i]);
    }
    
    for(int i=0; i<n; i++){
        for(int j=0; j<n; j++){
            for(int k=0; k<2; k++){
                c_x = i+X[k];
                c_y = j+Y[k];
                if(c_x >= n || c_y >= n)
                    continue;
                
                if(map[i][j] != map[c_x][c_y]){
                    int temp = map[i][j];
                    map[i][j] = map[c_x][c_y];
                    map[c_x][c_y] = temp;
                    
                    int cnt = find_candy(n);
                    if(cnt > max_candy)
                        max_candy = cnt;
                    
                    temp = map[i][j];
                    map[i][j] = map[c_x][c_y];
                    map[c_x][c_y] = temp;
                }
            }
        }
    }
    
    printf("%d\n", max_candy);
}