코딩테스트/프로그래머스

[그리디] 마법의 엘리베이터

박소민 2024. 1. 11. 20:40
마법의 엘리베이터
 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

  • 풀이
    • 5 기준으로 나눠야 한다는 것까지 생각하고 풀이 봄
    • 10으로 나눈 나머지로 1의 자리에서 갯수만 더해주면서 하나씩 앞으로 이동
    • 5인 경우
      • 55 이상부터는 10씩 더해주다가 100을 빼는게 이득
      • 45 이하는 그냥 10씩 빼주는게 이득
def solution(storey):
    answer = 0

    while storey:
        remainder = storey % 10
        # 6 ~ 9
        if remainder > 5:
            answer += (10 - remainder)
            storey += 10
        # 0 ~ 4
        elif remainder < 5:
            answer += remainder
        # 5
        else:
            if (storey // 10) % 10 > 4:
                storey += 10
            answer += remainder
        storey //= 10

    return answer