[백준] 1107: 리모컨 / Python / 브루트 포스

리모컨

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

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

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

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

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

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

나의 풀이

• N번 채널로 이동하는 방법이 2가지 있다는 것 까지는 잘 접근했다.
  - 방법 1. 현재 있는 100번 채널에서 + 혹은 - 버튼만 눌러서 N번으로 이동
  - 방법 2. (고장나지 않은) 숫자 버튼을 누른 후 + 혹은 - 버튼을 눌러 N번으로 이동
• 처음에는 방법 2를 구현할 때 바로 최적해를 찾으려고 했었다. 고장나지 않은 숫자 버튼들로 N과 가장 가까운 숫자를 만드려고 했으나 오히려 고려해야 할 케이스가 너무 많았다.
• 그렇게 풀 필요 없이 (누를 수 있는) 모든 채널로부터 N번으로 이동하는 경우를 비교하는 브루트 포스를 진행하면 된다.
• 백준 채점 시 EOFError가 자꾸 발생해서 뭔가 했더니, 고장난 버튼이 0일 때 입력값이 없는데 입력값을 받아오려고 해서 생기는 문제였다. 예외 처리를 꼼꼼하게 해주자.

unbroken = {i for i in range(10)}  

N = int(input()) 
M = int(input()) 
if M != 0:
    unbroken -= set(map(int, input().split()))  

# N번 채널로 이동하는 방법 2가지
# 1. 현재 있는 100번 채널에서 + 혹은 -만 눌러서 이동하기
result = abs(N - 100)

# 2. 고장난 숫자 버튼을 제외하고 숫자 버튼을 누른 후 + 혹은 -를 눌러 이동하기
# 입력 범위는 500,000번까지이지만 500,000보다 큰 숫자에서 N번으로 이동할 수도 있으므로 1,000,000까지 탐색
for channel in range(1000001):
    for idx, num in enumerate(str(channel)):
        # 고장난 숫자 버튼이 포함되면 break
        if int(num) not in unbroken:
            break
    # 고장난 버튼 없이 모두 누를 수 있다면 
    else:
        # (현재 누른 숫자 개수 + N번 채널로 이동하기 위해 눌러야 하는 + 혹은 - 횟수)와 minimal 값을 비교하여 갱신
        result = min(result, len(str(channel)) + abs(channel - N))
print(result)