백준 1107 [리모컨] 파이썬

https://www.acmicpc.net/problem/1107

1107번: 리모컨

# 백준 1107 리모컨

from itertools import product
num = input()
want = list(map(int, num))
l = len(want)
num = int(num)
n = int(input())
if n == 0:
    use = [i for i in range(10)]
else:
    delbtn = list(map(int, input().split()))
    use = list((set([i for i in range(10)]))-set(delbtn))
    use.sort()
if use:
    if not (len(use) == 1 and use[0] == 0):
        if use[0] == 0:
            big = int(str(use[1])+str(use[0])*l)
        else:
            big = int(str(use[0])*(l+1))
        if l != 1:
            small = int(str(use[-1])*(l-1))
        else:
            small = -1000000
        answer = min(abs(num-100), l-1+(num-small), l+1+(big-num))
    else:
        answer = min(abs(num-100), 1+num)
else:
    answer = abs(num-100)

use = list(map(str, use))
data = product(use, repeat=l)
for p in data:
    val = int(''.join(p))
    answer = min(answer, abs(val-num)+l)

print(answer)

 n 번째 숫자로 가는 가장 짧은 횟수는

min ( 위아래 버튼만 움직이기, 한자리가 더 긴 수에서 아래 버튼만 움직이기, 한자리가 더 작은 수에서 위 버튼만 움직이기, 사용가능한 모든 숫자로 이동한 뒤, 위 아래로 움직이기) 이므로,

위의 경우를 모두 따져주면 된다,

 

되게 간단한 문제라고 생각했는데... 어렵다. 맞추긴 했지만, 코드가 너무 지저분하다...

 

++ from itertools import product 에 대해 알아두자

++ 코드를 잘짜자