[BOJ S2][C++] 백준 3085번: 사탕 게임

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

3085번: 사탕 게임

 

문제

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

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

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

입력

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

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

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

출력

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

 

풀이

// Authored by : seondal
// Co-authored by : -

// #include <bits/stdc++.h>
#include <iostream>
#include <vector>

using namespace std;

int n;
vector<vector<char>> candy;

int dx[] = {0, 1};
int dy[] = {1, 0};

int countCandy() {
    // 가장 긴 연속 사탕의 갯수 반환 함수
    int max_candy = 0;
    
    // i번째 행의 가장 긴 연속 갯수 구하기
    for(int i=0; i<n; i++) {
        char prev = candy[i][0];
        int cnt = 1;

        for(int j=1; j<n; j++) {
            if(prev == candy[i][j]) { // 연속
                cnt++;
                max_candy = max(max_candy, cnt);
            } else { // 불연속
                prev = candy[i][j];
                cnt = 1;
            }
        }
    }
    
    // i번째 열의 가장 긴 연속 갯수 구하기
    for(int j=0; j<n; j++) {
        char prev = candy[0][j];
        int cnt = 1;

        for(int i=1; i<n; i++) {
            if(prev == candy[i][j]) {
                cnt++;
                max_candy = max(max_candy, cnt);
            } else {
                prev = candy[i][j];
                cnt = 1;
            }
        }
    }
    
    return max_candy;
}

int solution() {
    int ans = countCandy();
    
    // 모든 사탕에 대해서
    for(int i=0; i<n; i++) {
        for(int j=0; j<n; j++){
            
            // 오른쪽이나 아래쪽 사탕과 교환
            for(int dir=0; dir<2; dir++) {
                int x = i + dx[dir];
                int y = j + dy[dir];
                
                if(x>=n || y>=n)
                    continue;
                if(candy[i][j] == candy[x][y])
                    continue;
                
                swap(candy[i][j], candy[x][y]);
                ans = max(ans, countCandy());
                swap(candy[i][j], candy[x][y]); // 연속갯수 저장해주고 되돌리기
            }
        }
    }
    
    return ans;
}

int main() {
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    
    cin >> n;
    candy.assign(n, vector<char>(n));
    for(int i=0; i<n; i++)
        for(int j=0; j<n; j++)
            cin >> candy[i][j];
    
    cout << solution();
    
    return 0;
}

/*
 */