[ BOJ ] 1107 : 리모컨 ( GOLD 5 ) / Python

문제

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

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

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

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

입력

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

출력

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

풀이 과정

이동하려고 하는 채널 N은 0부터 500,000 까지 이므로, 숫자 버튼을 이용해 누르는 횟수는 최대 6회이다.

최소 버튼 횟수의 목적은, 100 채널에서 + - 버튼을 이용해 N으로 이동하는 것이 더 빠른지 혹은 숫자 버튼을 이용해 최대한 채널 N에 가까워진 후 + - 버튼을 이용하는 것이 더 빠른지를 재는 것이다.

백트래킹을 이용해 고장난 수를 사용하지 않으면서 N과 가장 가까운 수를 찾았다.

 

이 과정에서 N이 3자리 수일 때 가장 가까운 수는 3자리가 아닐 수도 있으며 (2자리일 수도, 4자리일 수도 있다.) 버튼을 누르는 횟수도 같이 고려해야 한다. (N과 차이가 동일한 3자리 수와 4자리 수가 있으면, 3자리 수가 버튼을 1회 덜 누른다.)

반례가 많은 문제이니 신경써서 구현한다.

import sys
input = sys.stdin.readline

n = int(input().rstrip())
m = int(input().rstrip())
arr = [0 for _ in range(10)]
if m != 0: 
    a = list(map(int, input().rstrip().split()))
    for i in a:
        arr[i] = 1
min_diff = -1

now = []
def backtracking():
    global min_diff

    if len(now) != 0:
        nint = int(''.join(map(str, now)))
        if min_diff == -1 or abs(n-min_diff) + len(str(min_diff)) > abs(n-nint) + len(str(nint)):
            min_diff = nint
    if len(now) == len(str(n)) + 1: return
    
    for i in range(10):
        if len(now) == 1 and now[-1] == 0: return
        if arr[i] == 1: continue

        now.append(i)
        backtracking()
        now.pop()

backtracking()

if min_diff != -1 and abs(n-100) > abs(n-min_diff) + len(str(min_diff)): print(abs(n-min_diff) + len(str(min_diff)))
else: print(abs(n-100))