백준 온라인 저지, 그리디 / 1107번: 리모컨 (파이썬 / 백준 골드문제)

2021. 9. 3. 00:12알고리즘/그리디

728x90
반응형

문제

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

문제 정의

수빈이는 TV를 보고 있다. 수빈이는 채널을 돌리려고 했지만, 버튼을 너무 세게 누르는 바람에, 일부 숫자 버튼이 고장났다.
리모컨에는 버튼이 0부터 9까지 숫자, +와 -가 있다. +를 누르면 현재 보고있는 채널에서 +1된 채널로 이동하고, -를 누르면 -1된 채널로 이동한다. 채널 0에서 -를 누른 경우에는 채널이 변하지 않고, 채널은 무한대 만큼 있다.
수빈이가 지금 이동하려고 하는 채널은 N이다. 어떤 버튼이 고장났는지 주어졌을 때, 채널 N으로 이동하기 위해서 버튼을 최소 몇 번 눌러야하는지 구하는 프로그램을 작성하시오. 
수빈이가 지금 보고 있는 채널은 100번이다.


입력

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


출력

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


예제 입력 1

5457
3
6 7 8

예제 출력 1

6

접근 방법

- 경우의 수 1. 버튼을 누를 수 있는 모든 채널에 대해 주어진 N과 가장 차이가 적은 것을 뽑아, 버튼을 누른 횟수 + N과의 차이 합을 출력한다.
- 경우의 수 2. 단, 경우에 따라 +와 -버튼만을 사용해 값을 출력하는 것이 더 빠른 경우가 존재한다. 따라서 +와 -만을 사용해 값을 출력하는 경우(abs(n-100))와 위의 방법을 사용하는 경우를 모두 고려해 가장 값이 작은 것을 출력한다.
- ex) N이 98 ~ 103인 경우 +와 -를 누르는 게 가장 적게 누르는 방법이기에 N이 위의 범위에 속할 경우 abs(n-100)을 출력한다.
- 최대 연산횟수: 각 자리수 마다 최대 10번씩 경우의 수가 생기는 것이므로 10 ** 자리수 만큼의 경우의 수가 발생한다. 따라서 약 1110만번의 연산횟수가 소요된다.(N보다 한자리수 높은 숫자의 경우의 수, )

코드

# https://www.acmicpc.net/problem/1107
# 접근 방법
# 경우의 수 1. 버튼을 누를 수 있는 모든 채널에 대해 주어진 N과 가장 차이가 적은 것을 뽑아, 버튼을 누른 횟수 + N과의 차이 합을 출력한다.
# 경우의 수 2. 단, 경우에 따라 +와 -버튼만을 사용해 값을 출력하는 것이 더 빠른 경우가 존재한다. 따라서 +와 -만을 사용해 값을 출력하는 경우(abs(n-100))와 위의 방법을 사용하는 경우를 모두 고려해 가장 값이 작은 것을 출력한다.
# ex) N이 98 ~ 103인 경우 +와 -를 누르는 게 가장 적게 누르는 방법이기에 N이 위의 범위에 속할 경우 abs(n-100)을 출력한다.
# 최대 연산횟수: 각 자리수 마다 최대 10번씩 경우의 수가 생기는 것이므로 10 ** 자리수 만큼의 경우의 수가 발생한다. 따라서 약 1110만번의 연산횟수가 소요된다.(N보다 한자리수 높은 숫자의 경우의 수, )
from itertools import product
n = int(input())
m = int(input())
Btn = [x for x in range(10)]
if m:
    unableBtn = list(map(int, input().split()))
    ableBtn = [str(x) for x in Btn if x not in unableBtn]
else:
    ableBtn = [str(x) for x in range(10)]
result = abs(n-100) # 값 초기화 (경우의 수 2)


channel = 500000
r = len(str(n))
if ableBtn:
    if len(ableBtn) >= 2 and ableBtn[0] == '0':
        num = ableBtn[1] + (ableBtn[0] * r)
        channel = min(channel, abs(n - int(num))+r+1)
    
    else:
        num = ableBtn[0] * (r+1)
        channel = min(channel, abs(n - int(num))+r+1)

    for data in list(product(ableBtn, repeat=r)):
        channel = min(channel, abs(n - int(''.join(list(data))))+r)
    if r >= 2:
        channel = min(channel, abs(n - int(''.join(list(product(ableBtn, repeat=r-1))[-1])))+r-1)
    result = min(result, channel)
print(result)
728x90
반응형