[Python][Gold 5][1107] 리모컨

|    문제

수빈이는 TV를 보고 있다. 수빈이는 채널을 돌리려고 했지만, 버튼을 너무 세게 누르는 바람에, 일부 숫자 버튼이 고장났다.

리모컨에는 버튼이 0부터 9까지 숫자, +와 -가 있다. +를 누르면 현재 보고있는 채널에서 +1된 채널로 이동하고, -를 누르면 -1된 채널로 이동한다. 채널 0에서 -를 누른 경우에는 채널이 변하지 않고, 채널은 무한대 만큼 있다.

수빈이가 지금 이동하려고 하는 채널은 N이다. 어떤 버튼이 고장났는지 주어졌을 때, 채널 N으로 이동하기 위해서 버튼을 최소 몇 번 눌러야하는지 구하는 프로그램을 작성하시오.

수빈이가 지금 보고 있는 채널은 100번이다.

 

|    입력

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

 

|    출력

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

 

 

---

정답 코드

import sys, bisect
from itertools import combinations
input = sys.stdin.readline

class Solution:
    def remote(self):
        n = input().rstrip()
        m = int(input())
        broken = tuple(map(int, input().split()))
        # +, - 버튼으로만 움직였을 때
        first_cnt = abs(100-int(n))
        # n의 범위(500_000) 보다 큰 숫자로 설정하여 브루트 포스
        for num in range(1_000_000):
        	# 고장난 버튼이 있는 경우
            for i in str(num):
                if int(i) in broken:
                    break
            # 고장난 버튼이 없는 경우 최소 비교
            else:
                first_cnt = min(first_cnt, len(str(num)) + abs(int(n) - num))
                
        print(first_cnt)


if __name__ == "__main__":
    s = Solution()
    s.remote()

 

 

|    비고

처음에 bisect 썼다가 안돼서 여러 경우로 나누고 그랬다. 왠지 이렇게 사고하는 게 아닌 거 같아서 구글링을 해봤는데 웬걸.. 정말 단순한 사고로 해결할 수 있었다. 요즘 어려운 문제도 풀다보니(?) 사고가 점점 꼬이는 것 같다.

예를 들어, 아래와 같은 입력값이 있다면 5번->4번->5번->(7번은 고장난 버튼이니)5번 -> +버튼 -> +버튼 이렇게 누를 수 있으므로 len('5455') + abs(5457-5455)로 계산할 수 있으며 최소 출력값은 6이 된다.

5457
3
6 7 8