BOJ 3085. 사탕 게임

🙇‍♀️[Silver II] 사탕 게임 - 3085

문제 링크

성능 요약

메모리: 2020 KB, 시간: 36 ms

분류

브루트포스 알고리즘, 구현

제출 일자

2023년 12월 22일 13:54:56

문제 설명

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

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

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

입력

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

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

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

출력

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

🚀풀이

내가 생각한 이 문제의 해결 로직은 다음과 같다.

1. 색이 서로 다른 인접한 사탕을 고른다. -> 하나의 사탕 당 상하좌우 모두 탐색
2. 위치를 변경한다.
3. 최대의 길이를 찾는다.
4. 보드의 모든 경우에서 위의 행위를 반복한다.

위의 로직대로 함수를 구현하면 아래와 같다.

int dy[] = { -1, 0, 1, 0 };
int dx[] = { 0, 1, 0, -1 };
void solve()
{
	int res = 0;

	for (int y = 0; y < n; ++y)
	{
		for (int x = 1; x < n; ++x)
		{
			for (int i = 0; i < 4; ++i)
			{
				int ny = y + dy[i];
				int nx = x + dx[i];

				if (ny < 0 || nx < 0 || ny >= n || nx >= n)
					continue;
				if (board[y][x] == board[ny][nx])
					continue;
				swap(board[y][x], board[ny][nx]);
				res = max(res, check_lis());
				swap(board[y][x], board[ny][nx]);
			}
		}
	}

	cout << res << '\n';
}

첫 이중 for문은 보드의 모든 위치를 탐색하는 것이고 마지막 for문은 상하좌우를 탐색하는 것이다.
swap을 해서 위치를 변경해주었고, check_lis함수를 만들어서 최대 길이를 찾았다.
문제에서 여러번 위치 변경이 안되므로 원상 복구를 시킨다.

최대 길이를 찾는 함수는 아래와 같이 구현하였다.

int check_lis()
{
	int res = 0, cnt = 0;
	char ch;

	for (int y = 0; y < n; ++y)
	{
		ch = board[y][0];
		cnt = 1;

		for (int x = 1; x < n; ++x)
		{
			if (ch == board[y][x])
			{
				cnt++;
				res = max(res, cnt);
			}
			else
			{
				cnt = 1;
				ch = board[y][x];
			}
		}
	}

	for (int x = 0; x < n; ++x)
	{
		ch = board[0][x];
		cnt = 1;

		for (int y = 1; y < n; ++y)
		{
			if (ch == board[y][x])
			{
				cnt++;
				res = max(res, cnt);
			}
			else
			{
				cnt = 1;
				ch = board[y][x];
			}
		}
	}

	return res;
}

좀 더 나은 코드가 있을거같은데 잘 생각이 안나서 있는 그래도 풀었다.

행과 열 모두 생각해야하므로 첫 이중 for문은 행을 탐색하여 최대 길이를 찾고, 그 뒤는 열을 탐색하여 최대 길이를 찾는다.

문자열이 달라질 때, cnt가 1부터 시작해야한다.
이거 헷갈려서 5분은 더 날린듯?

🚀전체 코드

#define _CRT_SECURE_NO_WARNINGS

#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include<iostream>
#include <fstream>
#include <vector>
#include <algorithm>

using namespace std;

int n, res = 0;
vector<vector<char>> board;

void setting()
{
	cin >> n;
	board = vector<vector<char>>(n, vector<char>(n));

	for (int y = 0; y < n; ++y)
		for (int x = 0; x < n; ++x)
			cin >> board[y][x];
}

// 문제의 해결 로직
// 1. 색이 서로 다른 인접한 사탕을 고른다. -> 하나의 사탕 당 상하좌우
// 2. 위치를 변경한다.
// 3. 최대 길이 수열을 찾는다.
// 4. 보드의 모든 경우에서 위의 행위를 반복한다.
int check_lis()
{
	int res = 0, cnt = 0;
	char ch;

	for (int y = 0; y < n; ++y)
	{
		ch = board[y][0];
		cnt = 1;

		for (int x = 1; x < n; ++x)
		{
			if (ch == board[y][x])
			{
				cnt++;
				res = max(res, cnt);
			}
			else
			{
				cnt = 1;
				ch = board[y][x];
			}
		}
	}

	for (int x = 0; x < n; ++x)
	{
		ch = board[0][x];
		cnt = 1;

		for (int y = 1; y < n; ++y)
		{
			if (ch == board[y][x])
			{
				cnt++;
				res = max(res, cnt);
			}
			else
			{
				cnt = 1;
				ch = board[y][x];
			}
		}
	}

	return res;
}

int dy[] = { -1, 0, 1, 0 };
int dx[] = { 0, 1, 0, -1 };
void solve()
{
	int res = 0;

	for (int y = 0; y < n; ++y)
	{
		for (int x = 1; x < n; ++x)
		{
			for (int i = 0; i < 4; ++i)
			{
				int ny = y + dy[i];
				int nx = x + dx[i];

				if (ny < 0 || nx < 0 || ny >= n || nx >= n)
					continue;
				if (board[y][x] == board[ny][nx])
					continue;
				swap(board[y][x], board[ny][nx]);
				res = max(res, check_lis());
				swap(board[y][x], board[ny][nx]);
			}
		}
	}

	cout << res << '\n';
}

int main()
{
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	//freopen("input.txt", "rt", stdin);

	setting();
	solve();

	return 0;
}