마법의 엘리베이터

문제 설명

마법의 세계에 사는 민수는 아주 높은 탑에 살고 있습니다. 탑이 너무 높아서 걸어 다니기 힘든 민수는 마법의 엘리베이터를 만들었습니다. 마법의 엘리베이터의 버튼은 특별합니다. 마법의 엘리베이터에는 -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

입출력 예1

Input : 16
Output : 6

입출력 예2

Input: 2554
Output : 16

문제풀이

이 문제는 설명에 답이 나와있다. 16인 경우 엘리베이터를 1만큼 6번 내려간 다음 10만큼 1번 내려가는 것보다 1만큼 4번 올라간 다음에 10만큼 2번 내려가면 더 편하다고 한다. 즉, storey를 10으로 나눴을 때 나머지가 6이상인 경우에는 무조건 올라가는 것이 더 좋다. 그리고 5이하인 경우에는 내려가는 것이 더 좋다. 그러나 딱 5인 경우에 대해서 문제가 발생한다. 숫자 95를 생각해보자. 5이하인 경우 내려가는 것을 생각하게 된다면 95에서 1만큼 5번 내려가고, 10만큼 1번 올라가고, 100만큼 1번 내려가면 총 7번이 든다. 그러나 1만큼 5번 올라가고, 100만큼 1번 내려가면 6번으로 더 적게 든다. 즉, 나머지가 5인 경우에 앞의 숫자에 따라 달라진다. 나머지가 딱 5인 경우 앞의 숫자가 5이상인 경우에는 올라가는 것이 더 이득이다. 그래서 아래와 같이 코드를 설계했다.

코드

def solution(storey):
    answer = 0

    while storey:
        num = storey % 10

        if num > 5:  # 5보다 큰 경우
            storey += 10 - num
            answer += 10 - num
        elif num == 5 and (storey // 10) % 10 >= 5:  # 5이면서 앞의 자리 숫자가 5와 크거나 같은 경우
            storey += 10 - num
            answer += 10 - num
        else:  # 5 이하인 경우
            answer += num

        storey //= 10

    return answer

solution(16)
solution(2554)
solution(95)