알고리즘 코드카타 117 - 마법의 엘리베이터

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

 

처음에는 간단한 문제라고 생각했다. 단순하게 현재 층의 각 자릿수를 5 이하면 내리고 그만큼 마법의 돌을 사용하고 6이상이면 올리고 10에서 뺀 만큼 마법의 돌을 사용하고 한개 더 사용하면 된다고 생각했기 때문이다.(다음 자릿수에서 한자리를 더 갔으므로)

 

def solution(storey):
    answer = 0
    digits = [int(digit) for digit in str(storey)]
    
    for digit in digits:
        if digit <= 5:
            answer += digit
        else:
            answer += 11-digit
        
    return answer

 

그래서 처음으로 작성한 코드는 위와같았다. 단순하게 현재 층수를 숫자단위로 분리해서 0과 10중 가까운 쪽으로 내림, 올림 하고 올렸을 경우에는 다음 층수를 고려해서 1을 추가로 더해주었다.

 

하지만 내가 고려하지 못한 점이 있었다. 현재 층이 5단위의 층이라면 일반적인 경우에는 내림을 하는 것이 마법의 돌을 덜사용하지만 다음 10단위의 층이  5보다 큰 경우에는 올림을 하는 것이 마법의 돌을 덜사용하기 때문이다. (다음 단위의 층에서 어차피 올림을 할 것이기 때문에 돌을 하나 더 사용해야한다는 점이 상쇄된다.)

 

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

 

그래서 코드를 다시 작성했다.

각 자릿수를 리스트 형태로 분리하는 대신에 1의 자리부터 현재층수를 조사해서 5보다 크다면 무조건 올라가는게 유리하므로 올라간다.

5인경우 다음 10의 자리의 층수를 고려하여 5보다 큰 경우 어차피 다음 단위의 층에서 올라가는 것을 선택할 것이므로 올림을 선택한다. 추가로, 다음 단위의 층수가 정확히 5인 경우는 내려가는 것과 차이가 나지 않지만 가장 앞자리의 층수가 5인 경우를 고려해서 올라가는 것을 선택한다.

마지막 else는 5보다 작은 경우로 항상 내려가는것이 유리하므로 내려가도록한다.

 

그리고 반복문의 마지막에 storey = storey//10를 입력해서 1의 자리를 삭제해주도록 하면 완료!

성공