[BOJ] 1107 리모컨 바로가기
수빈이는 TV를 보고 있다. 수빈이는 채널을 돌리려고 했지만, 버튼을 너무 세게 누르는 바람에, 일부 숫자 버튼이 고장났다.
리모컨에는 버튼이 0부터 9까지 숫자, +와 -가 있다. +를 누르면 현재 보고있는 채널에서 +1된 채널로 이동하고, -를 누르면 -1된 채널로 이동한다. 채널 0에서 -를 누른 경우에는 채널이 변하지 않고, 채널은 무한대 만큼 있다.
수빈이가 지금 이동하려고 하는 채널은 N이다. 어떤 버튼이 고장났는지 주어졌을 때, 채널 N으로 이동하기 위해서 버튼을 최소 몇 번 눌러야하는지 구하는 프로그램을 작성하시오.
수빈이가 지금 보고 있는 채널은 100번이다.
첫째 줄에 수빈이가 이동하려고 하는 채널 N (0 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 고장난 버튼의 개수 M (0 ≤ M ≤ 10)이 주어진다. 고장난 버튼이 있는 경우에는 셋째 줄에는 고장난 버튼이 주어지며, 같은 버튼이 여러 번 주어지는 경우는 없다.
첫째 줄에 채널 N으로 이동하기 위해 버튼을 최소 몇 번 눌러야 하는지를 출력한다.
cases = set()
for i in range(1,7):
cases = cases.union(set(int("".join(case)) for case in product(button, repeat=i)))
고장나지 않은 리모컨의 숫자 버튼으로 만들 수 있는 모든 경우의 수를 중복 순열(product
)을 통해 생성한다.
만약 고장나지 않은 리모컨의 숫자 버튼이 [0, 1, 2]
이라면 product([0,1,2], repeat=i)
를 통해 모든 경우의 수를 구한다.
i = 1) 0, 1, 2
i = 2) 10, 11, 12, 20, 21, 22
i = 3) 100, 101, 102, 110, 111, 112, 120, 121, 122, 200, 201, 202, 210, 211, 212
220, 221, 222
.
.
.
for case in cases:
move = len(str(case)) + abs(target - case)
if answer > move: answer = move
숫자 버튼으로 이동한 번호(case
)로 이동하기 위해 리모컨을 누른 횟수(len(str(case))
)와
목표 번호(target
)에 도달하기 위해 위 아래 버튼을 눌러야 하는 횟수(abs(target - case)
)를
더한 값 중 가장 작은 값을 찾는다.
# BOJ 1107 리모컨
# https://www.acmicpc.net/problem/1107
from sys import stdin
from itertools import product
def solution(target, button):
answer = abs(target - 100)
# 중복 순열을 통한 숫자 버튼을 이용해 이동할 수 있는 모든 경우
cases = set()
for i in range(1,7):
cases = cases.union(set(int("".join(case)) for case in product(button, repeat=i)))
# 최소 이동 횟수 갱신
for case in cases:
move = len(str(case)) + abs(target - case)
if answer > move: answer = move
return answer
# input
target = int(stdin.readline())
M = int(stdin.readline())
if M != 0: button = list(set([str(i) for i in range(10)]) - set(stdin.readline().strip().split()))
else: button = list(str(i) for i in range(10))
# response
response = solution(target, button)
print(response)