[Programmers] 프로그래머스 개미군단 Python

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

프로그래머스

 

def solution(hp):
    power = [5, 3, 1]
    count = 0

    for p in power:
        if hp >= p:
            c, d = divmod(hp, p)
            count += c
            hp = d
    return count