[C++] 1107: 리모컨

1107번: 리모컨 (acmicpc.net)

문제

수빈이는 TV를 보고 있다. 수빈이는 채널을 돌리려고 했지만, 버튼을 너무 세게 누르는 바람에, 일부 숫자 버튼이 고장났다.

리모컨에는 버튼이 0부터 9까지 숫자, +와 -가 있다. +를 누르면 현재 보고있는 채널에서 +1된 채널로 이동하고, -를 누르면 -1된 채널로 이동한다. 채널 0에서 -를 누른 경우에는 채널이 변하지 않고, 채널은 무한대 만큼 있다.

수빈이가 지금 이동하려고 하는 채널은 N이다. 어떤 버튼이 고장났는지 주어졌을 때, 채널 N으로 이동하기 위해서 버튼을 최소 몇 번 눌러야하는지 구하는 프로그램을 작성하시오.

수빈이가 지금 보고 있는 채널은 100번이다.

입력

첫째 줄에 수빈이가 이동하려고 하는 채널 N (0 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 고장난 버튼의 개수 M (0 ≤ M ≤ 10)이 주어진다. 고장난 버튼이 있는 경우에는 셋째 줄에는 고장난 버튼이 주어지며, 같은 버튼이 여러 번 주어지는 경우는 없다.

출력

첫째 줄에 채널 N으로 이동하기 위해 버튼을 최소 몇 번 눌러야 하는지를 출력한다.

예제 입력 1

5457
3
6 7 8

예제 출력 1

6

풀이

골드5인데 왜 안 풀리지 하고 낑낑대고 있었는데, 브루트포스 알고리즘. 즉 그냥 다 해 보는 문제였다! 목표 채널 제한이 500,000 이므로, 1,000,000 정도까지 다 검사하면서 버튼 누르는 횟수의 최소를 구하면 된다.

먼저 인풋을 받고, 횟수의 최소값을 찾는다.
0부터 백만까지 확인한다. 만약 자릿수에 고장난 버튼의 숫자가 있으면, continue하여 다음 숫자로 넘어간다.
채널의 각 자릿수에 고장난 버튼이 없다면, (목표 채널까지 +/-를 몇 번 눌러야 하는지) + (현재 채널의 자릿수)와 현재까지의 최소 횟수를 비교하여 최소 횟수를 갱신한다.
혹시 현재 채널 100에서 +/-만 눌러 가는 것이 더 적을 수 있으므로, 마지막에 비교해 준다.

주의! : 자릿수에 고장난 버튼을 체크할 때나, 자릿수를 체크할 때 while(channel > 0)을 사용하는데, 그럼 0이 제대로 체크되지 않아 틀렸다. 0은 따로 예외처리해 주었다.

코드

#include <iostream>
#include <vector>

using namespace std;

void	input_data(void);
int		remote_con(void);
int		find_min(void);
int		check_broken(int channel);
int		check_digit(int channel);

int goal, m;
vector<int> broken(0);

int main(void)
{
	input_data();
	cout << remote_con();

	return (0);
}

void input_data(void)
{
	cin >> goal;
	cin >> m;

	int i = -1;
	while (++i < m)
	{
		int j;
		cin >> j;
		broken.push_back(j);
	}

	return ;
}

int remote_con(void)
{
	int result;
	int min = find_min();
	result = abs(goal - 100);
	if (result > min)
		result = min;

	return (result);
}

int find_min(void)
{
	int channel = -1;
	int min_push = 1000000;

	while (++channel <= 1000000)
	{
		if (check_broken(channel) == 0)
			continue;

		int now_push = abs(channel - goal) + check_digit(channel);
		if (now_push < min_push)
			min_push = now_push;
	}

	return (min_push);
}

int check_broken(int channel)
{
	while (channel >= 0)
	{
		int units = channel % 10;
		int i = -1;
		while (++i < m)
		{
			if (units == broken[i])
				return (0);
		}
		channel /= 10;

		if (channel == 0)
			channel--;
	}

	return (1);
}

int check_digit(int channel)
{	
	int count = 0;

	if (channel == 0)
		return (1);

	while (channel > 0)
	{
		channel /= 10;
		count++;
	}

	return (count);
}

야호~~❤_❤