백준 1107번 : 리모컨(python)

문제

수빈이는 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

코드

import sys
n=int(input()) #1
m=int(input()) #2
button=['0','1','2','3','4','5','6','7','8','9']
min=500001
chk=0
if(m>0): #3 
    b=list(input().split())
else:
    b=[]
for i in b: #4
    button.remove(i)
for i in range(1000000): #5
    cnt=0
    for j in str(i):
        if(j not in button):
            cnt=0
            chk=1
            break
        else:
            cnt+=1
    if(chk==1):
        chk=0
        continue    
    if(min>cnt+abs(i-n)):
        min=cnt+abs(i-n)
if(min<abs(100-n)): #6
    print(min)
else:
    print(abs(100-n))

코드설명

1. 이동하려는 채널(n)을 입력받는다.

2. 부숴진 버튼의 개수(m)을 입력받는다.

3. m이 1개이상이면 부숴진 버튼을 입력받는다.

4. 온전한 버튼에서 부숴진 버튼을 빼준다.

5. 0부터 시작하여 모든수에 대하여 버튼을 누르고 +,-를 누르는 횟수의 최소값을 구하여준다.

5-1. 해당되는 숫자가 부숴지지않은 버튼에 해당되면 cnt를 증가시키면서 버튼을 누르는 횟수를 측정해주고, n값과 해당값을 뺀 절댓값을 취해줌으로써 +,-의 횟수를 측정하여준다.

5-2. 최소값 비교

6. 채널이 100부터 시작되기때문에 100에서 +,-만을 해주는 경우가 더 횟수가 적은 경우도 있다. 때문에 마지막에 100과 이동하려는 채널n까지의 횟수와도 비교하여 최종 최소값을 출력해준다. 

 

#5에서 범위를 1000000으로 한 이유는 500000까지만 한다면 +버튼을 누르는 경우에만 해당되기때문에 -를 하는 경우까지 넣어줘야되기에 1000000까지 해주었다.(500001~1000000까지의 채널이 나올수도있기때문)

#5에서 반복문을 돌리다보면 부숴진버튼의 값이 나올수도 있다. 이때 부숴진버튼은 누를수없기때문에 계산해주지않는다.(이중for문의 if문에 대한 설명)

느낀점 및 배운점

단순무식한 방법이 최선일때도 있다!

 

처음에는 중복순열을 이용하여 해결하려했는데 중복순열 함수를 만드는것이 쉽지 않았다. 꽤 많은 문제에서 보이는것을 보니 나중에 한번 따로 다뤄보아야겠다.