프로그래머스 python3 - 개미 군단

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

프로그래머스

 

문제 설명

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

---

제한사항

• hp는 자연수입니다.
• 0 ≤ hp ≤ 1000

---

입출력 예

 

입출력 예 설명

 

입출력 예 #1

• hp가 23이므로, 장군개미 네마리와 병정개미 한마리로 사냥할 수 있습니다. 따라서 5를 return합니다.

입출력 예 #2

• hp가 24이므로, 장군개미 네마리 병정개미 한마리 일개미 한마리로 사냥할 수 있습니다. 따라서 6을 return합니다.

입출력 예 #3

• hp가 999이므로, 장군개미 199 마리 병정개미 한마리 일개미 한마리로 사냥할 수 있습니다. 따라서 201을 return합니다.

문제 접근

1. hp를 5로 나눈 나머지를 가지고 다시 3으로 나눈 후, 그 다음 3으로 나눈 나머지를 가지고 1로 나누기

2. 이 과정에서 나온 몫들을 다 더하면 result

 

최종 코드

→ 단순하게 사칙연산으로만 계산해서 좀 지저분함

def solution(hp):
    n = int(hp / 5)
    m = int((hp-5*n)/3)
    k = int((hp-5*n-3*m)/1)
    return n+m+k

 

 

또 다른 풀이

→ 몫이나 나머지를 나타내는 사칙연산으로 더 간단하게 풀기

def solution(hp):
    n = hp // 5   #몫
    hp %= 5    #5로 나눈 나머지를 남김
    m = hp // 3
    hp %= 3
    k = hp // 1
    return n+m+k