[Baekjoon] 1107번 리모컨

문제

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

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

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

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

입력

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

출력

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

예제

Example 1:

Input: 
5457
3
6 7 8
Output: 
6

Example 2:

Input:
100
5
0 1 2 3 4
Output:
0

Example 3:

Input:
500000
8
0 2 3 4 6 7 8 9
Output:
11117

조건

시간 제한 : 2초

메모리 제한 : 256 MB

풀이과정

풀이 1

반례가 많은 문제. 브루트포스 방식으로 해결하였다. 채널이 0이 되는 경우도 고려하여야 한다.

import sys
n = int(input())

length = len(str(n))

m = int(input())

broken = []
if m != 0:
    broken = list(map(int, input().split()))

# 100번에서 위 아래로 가는 경우
ans = abs(n-100)

if m == 10:
    print(ans)
    sys.exit()

# 채널 누르는 경우
def dfs(L):
    global ans
    if L >=2 and arr[0] == 0:
        return

    if L >= length-1:
        val = int("".join(list(map(str,arr))))
        ans = min(ans, len(str(val))+abs(n - val))

    if L == length+1:
        return
    for i in range(10):
        if not i in broken:
            arr.append(i)
            dfs(L+1)
            arr.pop()

for i in range(10):
    if not i in broken:
        arr = [i]
        dfs(1)
print(ans)

풀이 2

100만까지 일일이 판단하는 방식

n = int(input())
m = int(input())

broken = [False]*10

if m != 0:
    arr = list(map(int, input().split()))
    for i in arr:
        broken[i] = True
def check(i):
    cnt = 0
    if i == 0:
        if broken[i]:
            return 0
        else:
            return 1

    else:
        while(i > 0):
            if broken[int( i % 10)]:
                return 0
            cnt += 1
            i = i // 10
        return cnt

ans = abs(n-100)

for i in range(1000001):
    len_i = check(i)
    if len_i > 0:
        ans = min(ans, len_i+abs(n-i))

print(ans)