Lv0. 개미군단

[개미군단]

문제 설명

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

제한사항

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

---

풀이

내 풀이

def solution(hp):
    answer = 0
    five = 0
    three = 0
    rem = hp%5
    five = hp//5
    if rem%3 == rem:
        answer = five+rem
    else:
        three = 1
        answer = five+rem//3+rem%3
    return answer

사실 문제를 뜯어보면 5 3 1 로 최적의 조합 만드는거였는데 사실상 따지고 보면 그냥 제일 큰수로 나누고 나머지 또 나누고 하면 되는거였음 나 혼자 그냥 뻘짓하느라 코드도 거지같이 짰다

다른 풀이 1

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

이게 내 코드를 최적화하면 이런 형태가 됨 굳이 메모리 써가면서 변수 저장할 필요 없음

다른 풀이 2

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

divmod 완전히 잊어버리고 있었지 뭐야~!처음 return 값이 몫, 그다음이 나머지 값이다