Algorithm/Python
[programmers]python 120837 개미군단
beomzh
2024. 4. 15. 13:46
728x90
반응형
# 개미군단
# 개미군단이 사냥을 나감
# 장군개미는 5의 공격력을, 병정개미는 3의 공격력을 일개미는 1의 공격력이 있음
# 사냥감의 체력 hp가 매개변수로 주어질 때, 사냥감의 체력에 딱 맞게 최소한의 병력을 구성하려면 몇 마리의 개미가 필요한지를 return
def solution(hp):
answer = 0
# hp가 0이 될때까지반복
while hp !=0:
# 가장 적은 병력 구성을 위해 개미 나눔
if hp >= 5 :
hp= hp-5
answer+=1
elif hp <5 and hp >= 3:
hp-=3
answer+=1
elif hp<3:
hp -=1
answer+=1
return answer
print(solution(23)) # 5
728x90
반응형