[백준] 사탕 게임 (3085번)

문제

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

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

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

입력

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

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

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

출력

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

예제 입력 1 복사

5 YCPZY CYZZP CCPPP YCYZC CPPZZ

예제 출력 1 복사

4

---

브루트 포스 문제이다.

입력으로 들어온 배열을 무제한으로 뒤섞을 수 있는 줄 알고 쫄았는데, 딱 한번만 섞을 수 있어서 간단한 문제였다.

인접한 두 사탕을 교환한 후, 가장 길게 연속으로 된 사탕 수를 파악하고 다시 원래대로 바꾼다.

 

NxN이기 때문에 (N-1) * N * 2 번의 연산으로 계산이 가능하다.

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/BaekJoon/2020/%EC%82%AC%ED%83%95%20%EA%B2%8C%EC%9E%84%20(3085%EB%B2%88).cpp

chosh95/STUDY

 

C++ 코드

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

int N, res = 1;
char p[51][51];

void checkSum() {
	int tmp = 1;
	for (int i = 1; i <= N; i++) {
		tmp = 1;
		for (int j = 1; j <= N - 1; j++) {
			if (p[i][j] == p[i][j + 1])
				tmp++;
			else
				tmp = 1;
			res = max(res, tmp);
		}
	}

	for (int i = 1; i <= N; i++) {
		tmp = 1;
		for (int j = 1; j <= N - 1; j++) {
			if (p[j + 1][i] == p[j][i])
				tmp++;
			else
				tmp = 1;
			res = max(res, tmp);
		}
	}
}

int main()
{
	cin >> N;
	for (int i = 1; i <= N; i++) {
		string str;
		cin >> str;
		for (int j = 1; j <= N; j++) {
			p[i][j] = str[j - 1];
		}
	}

	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= N - 1; j++) {
			swap(p[i][j],p[i][j+1]);
			checkSum();
			swap(p[i][j], p[i][j+1]);

			swap(p[j][i], p[j+1][i]);
			checkSum();
			swap(p[j][i], p[j+1][i]);
		}
	}

	cout << res;
}