[프로그래머스][Lv.0] 개미 군단

문제

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

프로그래머스

 

내 풀이 A

def solution(hp):
    answer = 0
    a = hp // 5
    print('a: ', a)
    hp -= (a*5)
    b = hp // 3
    print('b: ', b)
    hp -= (b*3)
    c = hp // 1
    print('c: ', c)
    answer = a+b+c
    
    return answer

 

내 풀이 B

def solution(hp):
    answer = 0
    for power in [5, 3, 1]:
        ant = hp // power
        hp -= ant * power
        answer += ant
    
    return answer

위에서 푼 방법을 정리해 개미 종류가 늘어나도 손쉽게 확장할 수 있도록 변경해봤다.

 

다른 풀이

def solution(hp):
    answer = 0
    for ant in [5, 3, 1]:
        d, hp = divmod(hp, ant)
        answer += d
    return answer

divmod(a, b) = 몫, 나머지

a를 b로 나눈 값의 몫과 나머지를 출력하는 함수를 사용해서

나처럼 변하는 hp에 대해 나머지로 더 편하게 구하는 방법을 사용했음