[프로그래머스/Python] 마법의 엘리베이터

문제 정보

제목 : 마법의 엘리베이터

난이도 : Lv.2

사용 언어 : Python

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/148653

프로그래머스

 

문제 설명

마법의 세계에 사는 민수는 아주 높은 탑에 살고 있습니다. 탑이 너무 높아서 걸어 다니기 힘든 민수는 마법의 엘리베이터를 만들었습니다. 마법의 엘리베이터의 버튼은 특별합니다. 마법의 엘리베이터에는 -1, +1, -10, +10, -100, +100 등과 같이 절댓값이 10c (c ≥ 0 인 정수) 형태인 정수들이 적힌 버튼이 있습니다. 마법의 엘리베이터의 버튼을 누르면 현재 층 수에 버튼에 적혀 있는 값을 더한 층으로 이동하게 됩니다. 단, 엘리베이터가 위치해 있는 층과 버튼의 값을 더한 결과가 0보다 작으면 엘리베이터는 움직이지 않습니다. 민수의 세계에서는 0층이 가장 아래층이며 엘리베이터는 현재 민수가 있는 층에 있습니다.

마법의 엘리베이터를 움직이기 위해서 버튼 한 번당 마법의 돌 한 개를 사용하게 됩니다.예를 들어, 16층에 있는 민수가 0층으로 가려면 -1이 적힌 버튼을 6번, -10이 적힌 버튼을 1번 눌러 마법의 돌 7개를 소모하여 0층으로 갈 수 있습니다. 하지만, +1이 적힌 버튼을 4번, -10이 적힌 버튼 2번을 누르면 마법의 돌 6개를 소모하여 0층으로 갈 수 있습니다.

마법의 돌을 아끼기 위해 민수는 항상 최소한의 버튼을 눌러서 이동하려고 합니다. 민수가 어떤 층에서 엘리베이터를 타고 0층으로 내려가는데 필요한 마법의 돌의 최소 개수를 알고 싶습니다. 민수와 마법의 엘리베이터가 있는 층을 나타내는 정수 storey 가 주어졌을 때, 0층으로 가기 위해 필요한 마법의 돌의 최소값을 return 하도록 solution 함수를 완성하세요.

 

제한 사항

• 1 ≤ storey ≤ 100,000,000

입출력 예

storey	result
16	6
2554	16

 

입출력 예 설명

입출력 예 #1

• 문제 예시와 같습니다.

입출력 예 #2

• -1, +100이 적힌 버튼을 4번, +10이 적힌 버튼을 5번, -1000이 적힌 버튼을 3번 누르면 0층에 도착 할 수 있습니다. 그러므로 16을 return 합니다.

---

나의 풀이

문제를 읽다보니 그리디가 생각났습니다. storey가 0보다 크다는 조건을 만족하는 동안 반복하는 while루프를 이용했습니다. storey%10을 구해 일의 자리 숫자를 구합니다. 이것은 -,+ 중 어느 방향으로 갈지를 정하게 됩니다. btn이 5보다크면 층을 올라가는게 낫기 때문에 answer에 (10-btn)층 만큼 더하고 storey도 그만큼 더합니다. 반대로 5보다 작으면 층을 내려가는게 더 빠르기 때문에 answer에 btn층을 더하고 storey은 btn만큼 뺍니다. 만약 5라면 그 앞의 자릿수를 구해 비교해서 계산하여 일의 자리 숫자를 0으로 맞춥니다. 일의 자리 숫자만큼은 이동한 것이므로  storey //= 10을 합니다.

코드

def solution(storey):
    answer = 0
    while(storey>0) :
        btn = storey %10
        if(btn > 5) :
            answer += (10-btn)
            storey += (10-btn)
        elif(btn < 5) :
            answer += btn
            storey -= btn
        else :
            tmp = storey // 10
            if(tmp%10 >= 5) :
                answer += (10-btn)
                storey += (10-btn)
            else :
                answer += btn
                storey -= btn
        storey //= 10
    return answer