[백준-1107] 리모컨 / Python

1107번: 리모컨

import sys

input = sys.stdin.readline
enable = {str(i) for i in range(10)}
n = int(input())
m = int(input())
if m != 0:
    enable -= set(map(str, input().split()))

answer = abs(100 - n)

for num in range(1000001):
    num = str(num)
    for j in range(len(num)):
        if num[j] not in enable:
            break
        elif j == len(num) - 1:
            answer = min(answer, abs(n - int(num)) + len(num))

print(answer)

 

해설

부르트포스 알고리즘으로 해결 가능한 문제다.

+로 찾을 수 있는 최적과, -로 찾을 수 있는 최적을 고려해 탐색 범위는 1000000으로 잡아준다.

enable에는 망가진 버튼을 제외한 버튼들이 들어있는 리스트다.

사용할 수 있는 버튼의 조합 중에서

min(answer, abs(n - int(num)) + len(num))

을 만족하는 수가 답이다.