[알고리즘] BOJ 1107 리모컨 #Python

[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)