프로그래머스_ 개미 군단

문제 설명

개미 군단이 사냥을 나가려고 합니다. 개미군단은 사냥감의 체력에 딱 맞는 병력을 데리고 나가려고 합니다. 장군개미는 5의 공격력을, 병정개미는 3의 공격력을 일개미는 1의 공격력을 가지고 있습니다. 예를 들어 체력 23의 여치를 사냥하려고 할 때, 일개미 23마리를 데리고 가도 되지만, 장군개미 네 마리와 병정개미 한 마리를 데리고 간다면 더 적은 병력으로 사냥할 수 있습니다. 사냥감의 체력 `hp`가 매개변수로 주어질 때, 사냥감의 체력에 딱 맞게 최소한의 병력을 구성하려면 몇 마리의 개미가 필요한지를 return하도록 solution 함수를 완성해주세요.

 

Answer 1

• y = 5x1 + 3x2 + x3의 식을 토대로 코드를 짜보기 시작함
• 5로 나눈 몫과 5로 나눈 나머지를 3으로 나눈 몫과 나머지를 모두 더한 값을 반환

def solution(hp):
    return hp // 5 + (hp % 5 // 3) + ((hp % 5) % 3)

 

Answer 2

• sort() 함수를 사용해서 sides 리스트를 오름차순으로 정렬
• 최대 값은 배열의 마지막에 위치해 있으므로 sides[:2]로 슬라이싱을 사용해 나머지 지정
• 나머지 두 수의 합은 sum() 함수를 이용
• 최대 값이 나머지 두 수의 합보다 작으면 1 반환, 아니면 2 반환

def solution(hp):
    ga = hp % 5
    if ga:
        if ga % 3:
            return hp // 5 + ga // 3 + ga % 3
        else:
            return hp // 5 + ga // 3
    else:
        return hp // 5

 

Answer 3

• 같은 코드지만 가독성있게 조건을 나누어 표현
• 공격력을 5로 나누었을 때, 나머지가 있으면 계속 계산하고 없으면 5로 나누어 떨어지는 것이므로 몫 반환
• 5로 나눈 나머지가 3으로 나눈 나머지가 0이면, 3으로 나누어 떨어지므로 5로 나눈 몫과 5로 나눈 나머지를 3으로 나눈 몫을 더한 값 반환
• 3으로 나눈 나머지가 있으면, 5로 나눈 몫과 5로 나눈 나머지를 3으로 나눈 몫과 나머지를 모두 더한 값을 반환

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

---

divmod() 함수

두 개의 숫자를 인자로 받아서 첫 번째 숫자를 두 번째 숫자로 나눈 몫과 나머지를 튜플 형태로 반환

a, b = divmod(10,3)
print(a)
print(b)

#결과
3
1