2024. 7. 28. 18:00ㆍ코딩테스트 입문
https://school.programmers.co.kr/learn/courses/30/lessons/120837
문제 설명
개미 군단이 사냥을 나가려고 합니다. 개미군단은 사냥감의 체력에 딱 맞는 병력을 데리고 나가려고 합니다. 장군개미는 5의 공격력을, 병정개미는 3의 공격력을 일개미는 1의 공격력을 가지고 있습니다. 예를 들어 체력 23의 여치를 사냥하려고 할 때, 일개미 23마리를 데리고 가도 되지만, 장군개미 네 마리와 병정개미 한 마리를 데리고 간다면 더 적은 병력으로 사냥할 수 있습니다. 사냥감의 체력 hp가 매개변수로 주어질 때, 사냥감의 체력에 딱 맞게 최소한의 병력을 구성하려면 몇 마리의 개미가 필요한지를 return하도록 solution 함수를 완성해주세요.
제한사항
- hp는 자연수입니다.
- 0 ≤ hp ≤ 1000
입출력 예
hp | result |
23 | 5 |
24 | 6 |
999 | 201 |
입출력 예 설명
입출력 예 #1
- hp가 23이므로, 장군개미 네마리와 병정개미 한마리로 사냥할 수 있습니다. 따라서 5를 return합니다.
입출력 예 #2
- hp가 24이므로, 장군개미 네마리 병정개미 한마리 일개미 한마리로 사냥할 수 있습니다. 따라서 6을 return합니다.
입출력 예 #3
- hp가 999이므로, 장군개미 199 마리 병정개미 한마리 일개미 한마리로 사냥할 수 있습니다. 따라서 201을 return합니다.
최대한 적은 개수의 개미들로 사냥을 하는 문제입니다.
최소한의 병력을 구성하려면 가장 전투력이 높은 개미들로 구성하면 되지 않나 싶은데, 사냥감의 체력에 딱 맞는 병력을 데리고 나가려고 한다고 문제에 제시되어 있습니다.
따라서 이 문제는 장군개미 선발 -> 병정 개미 선발 -> 일개미 선발 순으로 병력 수를 구해야 합니다.
def solution(hp):
answer = 0
# 장군개미 카운트
cnt = hp // 5
hp -= cnt * 5
answer += cnt
# 병정개미 카운트
cnt = hp // 3
hp -= cnt * 3
answer += cnt
# 일개미 카운트
answer += hp
return answer
장군개미를 카운트 후 병정개미 카운트, 마지막으로 남은 값은 전부 일개미로 채우는 풀이입니다.
가장 간단하게 풀 수 있는 방법입니다.
이 방법 외에도 조금 생각해볼 수 있습니다.
결국 전투력을 제외하고 나서 남은 값들에 연산을 계속하는 게 주요 포인트인데, 이 문제에서는 연산은 오직 나눗셈 뿐입니다.
따라서 나눗셈을 활용하는 방법으로 문제를 풀 수 있습니다.
def solution(hp):
# 장군개미의 수: hp // 5
# 병정개미의 수: 장군개미 전투력을 제외한 값(hp % 5) // 3
# 일개미의 수: 장군, 병정개미 전투력을 제외한 값(hp % 5 % 3)
answer = hp // 5 + (hp % 5 // 3) + (hp % 5 % 3)
return answer
장군개미는 hp // 5만큼의 수를 갖습니다.
hp % 5가 결국 남은 전투력이 되기 때문에, 이 값을 3으로 나눈 몫이 병정개미의 수가 됩니다.
hp % 5 % 3은 장군개미도, 병정개미로도 나눌 수 없는 0~2 사이의 값입니다. 이 값들이 곧 일개미의 수입니다.
최소한의 병력을 딱 맞춰 가져가는 문제라서 사용할 수 있는 방법입니다.