백준 1107번 | 리모컨 - python

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

예제 입력 2

100
5
0 1 2 3 4

예제 출력 2

0

예제 입력 3

500000
8
0 2 3 4 6 7 8 9

예제 출력 3

11117

예제 입력 4

100
3
1 0 5

예제 출력 4

0

예제 입력 5

14124
0

예제 출력 5

5

예제 입력 6

1
9
1 2 3 4 5 6 7 8 9

예제 출력 6

2

예제 입력 7

80000
2
8 9

예제 출력 7

2228

힌트

예제 1의 경우 5455++ 또는 5459--

 

 

이번 문제는 부르트포스 알고리즘을

이용한 문제로

하나하나 다 돌려본는 것이

최고라고 생각한다

 

python의 경우에는 이러한 것이

itertools에 잘 짜여져있기 때문에

그걸 활용해보도록 하자.

 

특히, 리모컨의 숫자를 중복해서

뽑을 수 있기 떄문에

중복 조합으로

문제를 해결하면 될 것이다.

 

아래의 코드는 itertools 의 product를 이용한 풀이이다

 

import sys
from itertools import product

# 누른 버튼의 갯수를 측정하기 위한 함수
def TenBagger(n) -> int:
    for idx in range(len(n)):
        if n[idx] != '0':
            return len(n) - idx
    return 1


n = list(sys.stdin.readline().strip())
m = int(sys.stdin.readline())
# 고장난 버튼이 있는 경우에는 고장난 숫자를 빼주고, m이 0이면 0 ~ 9까지 숫자를 string 형식으로 저장
if m:
    buttons = list(set(map(str, range(10))) - set(map(str, sys.stdin.readline().split())))
else:
    buttons = list(map(str, range(10)))

# 우리가 보고 싶은 채널의 정수형
goal = int(''.join(n))
# 초기 채널번호가 100이므로 goal과 100의 차이를 미리 저장
answer = abs(goal - 100) if goal != 100 else 0

# 만약 9 에서 10으로 넘어가거나, 10에서 9로 넘어가는 경우가 최소일 수도 있어서,,,
for i in [-1, 0, 1]:
    # 채널의 갯수가 1개 이상이므로 0보다 클 때,
    if len(n) + i > 0:
        # itertools의 product를 이용해 중복 사용 가능하게 permutations
        for cwr in product(buttons, repeat=len(n) + i):
            # 구한 값을 정수형으로 변형
            channel = int(''.join(cwr))
            answer = min(abs(channel - goal) + TenBagger(cwr), answer)
print(answer)

 

이렇게 숫자들을 리스트에 넣고 풀이를 진행했는데

 

itertools를 사용할 떄 주의할 점은 가끔 시간초과도 뜨는 문제들도 있어서

 

걸리는 시간을 고려해서 사용하길 바랍니다.

 

 

해당 게시물이 도움이 되셨다면,