수빈이는 TV를 보고 있다. 수빈이는 채널을 돌리려고 했지만, 버튼을 너무 세게 누르는 바람에, 일부 숫자 버튼이 고장났다.
리모컨에는 버튼이 0부터 9까지 숫자, +와 -가 있다. +를 누르면 현재 보고있는 채널에서 +1된 채널로 이동하고, -를 누르면 -1된 채널로 이동한다. 채널 0에서 -를 누른 경우에는 채널이 변하지 않고, 채널은 무한대 만큼 있다.
수빈이가 지금 이동하려고 하는 채널은 N이다. 어떤 버튼이 고장났는지 주어졌을 때, 채널 N으로 이동하기 위해서 버튼을 최소 몇 번 눌러야하는지 구하는 프로그램을 작성하시오.
수빈이가 지금 보고 있는 채널은 100번이다.
첫째 줄에 수빈이가 이동하려고 하는 채널 N (0 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 고장난 버튼의 개수 M (0 ≤ M ≤ 10)이 주어진다. 고장난 버튼이 있는 경우에는 셋째 줄에는 고장난 버튼이 주어지며, 같은 버튼이 여러 번 주어지는 경우는 없다.
첫째 줄에 채널 N으로 이동하기 위해 버튼을 최소 몇 번 눌러야 하는지를 출력한다.
예제 1의 경우 5455++ 또는 5459--
그리디 알고리즘으로 접근하여 문제를 풀려고 했으나, 그렇지 않은 예외들이 존재하여 그리디 알고리즘으로는 풀 수 가 없다.
그래서 그냥 브루트 포스로 풀이하였다.
알고리즘 분류도 브루트 포스로 되어 있어서..
시작 cnt 는 현재 채널인 100 부터 목표 채널 까지 그냥 +/- 하는 경우.
0 부터 1000000 까지 가능한 번호 숫자 중, 해당 번호에서 +/- 하는 경우.
이렇게 두 가지 경우로 나눠서 문제를 풀이했다.
def solution(N, M, broken):
cnt = abs(N-100)
buttons = []
for i in range(10):
if i not in broken:
buttons.append(i)
if N == 100:
return 0
if not buttons:
return abs(100 - N)
for i in range(1000001):
num_list = list(map(int,str(i)))
valid = True
for n in num_list:
if n in broken:
valid = False
break
if valid:
cnt = min(cnt, len(num_list) + abs(N - i))
return cnt
import sys
lines = sys.stdin.readlines()
N, M, broken = 0, 0, []
for i in range(len(lines)):
if i == 0:
N = int(lines[i])
if i == 1:
M = int(lines[i])
if i == 2:
broken = list(map(int,lines[i].split()))
print(solution(N, M, broken))