2021. 9. 3. 00:12ㆍ알고리즘/그리디
문제
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)
'알고리즘 > 그리디' 카테고리의 다른 글
백준 온라인 저지, 그리디 / 2457번: 공주님의정원 (파이썬 / 백준 골드문제) (0) | 2021.11.09 |
---|---|
백준 온라인 저지, 그리디 / 2170번: 선긋기 (파이썬 / 백준 골드문제) (0) | 2021.09.07 |
백준 온라인 저지, 그리디 / 1689번: 겹치는선분 (파이썬 / 백준 골드문제) (0) | 2021.09.03 |
백준 온라인 저지, 그리디 / 1461번: 도서관 (파이썬 / 백준 골드문제) (0) | 2021.09.01 |
백준 온라인 저지, 그리디 / 13904번: 과제 (파이썬 / 백준 골드문제) (0) | 2021.09.01 |