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

문제

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

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

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

입력

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

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

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

출력

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

예제 입력 1

3
CCP
CCP
PPC

예제 출력 1

3

예제 입력 2

4
PPPP
CYZY
CCPY
PPCC

예제 출력 2

4

알고리즘 [ 접근 방법 ]

NxN 크기에 사탕을 넣고 색이 다른 인접한 두칸을 골라 서로 사탕을 교환한다. 모두 같은 색으로 이루어진 가장 긴 연속부분(열 또는 행)을 골라서 다 먹는데 그 최대 개수를 구하라는 문제이다. 시간 복잡도는 O(N^2)이고 N <=50이므로 전체 경우의 수가 몇개 안되니깐 브루트 포스를 이용해 풀 수 있다.

먼저 현재 구성된 사탕의 배열에서 먹을 수 있는 가장 긴 사탕개수를 구하는 함수( NumCandy( ) )를 만들어 놓는다.

그 다음 색깔이 서로 다른 두개의 사탕을 swap하고 가장 긴 사탕개수를 구하는 NumCandy를 실행시킨뒤 결과를 저장한다.

그 다음 다시 원래대로 사탕을 swap하고 위의 과정을 반복한다.

이때 행끼리의 swap과 열끼리의 swap을 구분해서 진행해준다.

풀이

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
string candy[50];
int test;
int result = 0;

int NumCandy() {
	int result = 1;
	for (int i = 0; i < test; i++) {
	int temp = 1;
		for (int j = 1; j < test; j++) {
			if (candy[i][j - 1] == candy[i][j]) temp++;
			else {
				result = max(temp, result);
				temp = 1;
			}
			result = max(temp, result);
		}
	}

	for (int i = 0; i < test; i++) {
		int temp = 1;
		for (int j = 1; j < test; j++) {
			if (candy[j][i] == candy[j - 1][i]) temp++;
			else {
				result = max(temp, result);
				temp = 1;
			}
		}
		result = max(temp, result);
	}

	return result;
}

int main() {
	int cnt = 0;
	cin >> test;

	int result = 0;

	for (int i = 0; i < test; i++) cin >> candy[i];

	for (int i = 0; i < test; i++) {
		for (int j = 1; j < test; j++) {
			swap(candy[i][j - 1], candy[i][j]);
			result = max(result, NumCandy());
			swap(candy[i][j - 1], candy[i][j]);

			swap(candy[j][i], candy[j - 1][i]);
			result = max(result, NumCandy());
			swap(candy[j][i], candy[j - 1][i]);
		}
	}

	cout << result << endl;


}