프로그래머스 - 개미 군단 / Python

문제설명

---

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

 

 

 

 

 

제한사항

---

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

 

 

 

 

 

입출력 예

---

hp	result
23	5
24	6
999	201

 

 

 

 

 

 

입출력 예 설명

---

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

 

 

 

 

 

정답

---

def solution(hp):
    answer = 0
    
    general_ant = hp // 5
    soldier_ant = (hp - (5*general_ant)) // 3
    ergate = (hp - (5*general_ant) - (3*soldier_ant)) // 1
    
    return general_ant + soldier_ant + ergate

 

장군개미와 병정개미, 일개미 수를 구하는 계산식을 사용하여 각각 출력된 결과를 다 합쳐 리턴한다.

 

 

general_ant = hp // 5

 

장군개미는 5의 공격력을 가지고 있다고 했으니 5로 나눠 몫을 구했다.

 

**

파이썬 연산자 중 " // " 은 정수의 몫을 구하는 산술연사자이다.

hp가 자연수라고 하였으니 사용가능.

 

 

soldier_ant = (hp - (5*general_ant)) // 3

 

병정개미 수 구하는 식.

몫만 구하는 식을 사용했기 때문에 남은 값을 다음으로 넘길 수 없다.

그래서 위에서 가져온 값에 다시 공격력을 곱해 나눈다.

출력은 각각의 값을 다 더해서 출력.

 

 

 ergate = (hp - (5*general_ant) - (3*soldier_ant)) // 1

 

일개미 구하는 식.

역시 몫만 구하는 식을 사용했기 때문에 사냥감의 체력값인 hp에서 장군개미 수와 병정개미 수를 제외하고 공격력 1로 나눈다.

 

 

 

 

끝!