[백준/c++] 3085 사탕게임

문제

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

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

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

입력

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

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

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

출력

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

구현해야할 조건

처음에는 4방향 전부를 건드려야하나 생각을 했지만, (0,0)에서 (n-1,n-1)까지 한칸씩 이동하면서 우측과 아래측을 바꿔주면 된다는 생각을 했고, 굳이 4방향 전부 건드릴 이유가 없었다. swap 함수를 사용해서 쉽게 바꿔주고, 내부 검사를 해서 결과를 만들었다.

code

#include<iostream>
#include<cstring>
#include<algorithm>

#define MAX 51
using namespace std;

const int dx[2] = {0,1};
const int dy[2] = {1,0}; // down, right

int n,result;
char board[MAX][MAX];
char tempBoard[MAX][MAX]; // 초반에 DFS로 풀려고 헀었다

int candy()
{
	// 바꿔주면 바꿔준대로 길이를 측정해보자 right;
	int cnt = 1;
	
	for(int i = 0; i < n; i++)
	{
		int temp = 1;
		
		for(int j = 1; j < n; j++)
		{
			if(board[i][j-1] == board[i][j])
				temp++;
			else // 같지 않으면 이전 길이는 끝 
			{
				cnt = max(cnt, temp);
				temp = 1;
			}
		}
		cnt = max(cnt,temp);
	}
	
	for(int i = 0; i < n; i++)
	{
		int temp = 1;
		
		for(int j = 0; j < n - 1; j++)
		{
			if(board[j+1][i] == board[j][i])
			{
				temp++;
			}
			else
			{
				cnt = max(cnt, temp);
				temp = 1;
			}
		}
		
		cnt = max(cnt, temp);
	}
	
	return cnt;
}

int main()
{
	cin >> n;
	
	for(int i = 0; i < n; i++)
	{
		for(int j = 0; j < n; j++)
		{
			cin >> board[i][j];
		}
	}
	
	result = 0;
	
	for(int i = 0; i < n; i++)
	{
		for(int j = 0; j < n; j++)
		{
			swap(board[i][j],board[i][j+1]);
			result = max(result, candy());
			swap(board[i][j],board[i][j+1]); // right
			
			swap(board[j][i], board[j+1][i]);
			result = max(result, candy());
			swap(board[j][i], board[j+1][i]); // down
		}
	}
	
	cout << result;
	return 0;
	
}