개미 군단

2024. 7. 28. 18:00코딩테스트 입문

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


문제 설명
개미 군단이 사냥을 나가려고 합니다. 개미군단은 사냥감의 체력에 딱 맞는 병력을 데리고 나가려고 합니다. 장군개미는 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 사이의 값입니다. 이 값들이 곧 일개미의 수입니다.

 

최소한의 병력을 딱 맞춰 가져가는 문제라서 사용할 수 있는 방법입니다.

'코딩테스트 입문' 카테고리의 다른 글

가위 바위 보  (0) 2024.07.30
모스부호 (1)  (0) 2024.07.29
순서쌍의 개수  (0) 2024.07.27
진료순서 정하기  (0) 2024.07.26
외계행성의 나이  (0) 2024.07.25