[백준/python] 1107번 : 리모컨

문제

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

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

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

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

 

입력

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

 

출력

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

 

정답

import sys
input = sys.stdin.readline

button = []

N = int(input())
M = int(input())
if M != 0:
    button.extend(list(map(str, input().split())))

answer = abs(N-100)    # +,-로만 이동 횟수

if 100 == N:    # 이동하려는 채널이 현재 있는 채널인 경우
    print(0)
    
elif M == 10: # 모든 숫자 버튼이 고장난 경우 : +,-로만 이동
    print(answer)

else : 
    for count in range(1000001):
        if N-count >= 0:
            ok = True
            for i in str(N-count):
                if i in button:
                    ok = False
                    break
            if ok == True:
                print(min(answer, count + len(str(N-count))))
                break
        
        ok = True
        for i in str(N+count):
            if i in button:
                ok = False
                break
        if ok == True:
            print(min(answer, count + len(str(N+count))))
            break

---

고장난 숫자가 없을 경우를 대비하기 위해

button 리스트를 우선 만들어 둔 후 M이 0이 아닐 때 입력을 받아 button에 추가한다.

 

이동하려는 채널 N이 현재 채널 100과 같은 경우 0을 출력하고

모든 숫자 버튼이 고장났을 경우 +와 -버튼만 조작할 수 있기 때문에 N과 100의 차이를 출력한다.

이외의 경우 +,- 버튼으로만 이동했을 때와 숫자 버튼을 이용한 이동 중 더 작은 값을 출력한다.

 

목표 채널 N에서 1씩 빼거나 더하여

해당 채널로 이동할 수 있는지 확인한 다음 

이동할 수 있다면 +,-로만 이동했을 경우와 비교하여 더 작은 값을 출력한다.