프로그래머스/Lv. 0(코딩테스트 입문)

[프로그래머스 코딩테스트] 개미 군단(Java)

Sigfriede 2023. 4. 1. 01:45

  문제 설명

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

 

  제한사항

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

 

  입출력 예

hp result
23 5
24 6
999 201
class Solution {
    public int solution(int hp) {
        int answer = 0;
        int general = 5;
        int soldier = 3;
        int worker = 1;
        
        answer = (hp / general) + (hp % general) / soldier + ((hp % general) % soldier) / worker;
        
        return answer;
    }
}

  저는 변수 사용을 선호해서 이번에도 마찬가지로 변수를 사용해서 풀었습니다. 간단한 문제여서 그런지, 오히려 변수 사용이 식의 가독성을 해치는 것도 같네요. 위의 식을 변수 없이 풀면 이런 식이 됩니다.

 

answer = (hp / 5) + (hp % 5) / 3 + ((hp % 5) % 3) / 1;

 

  문제는 최소한의 병력 사용을 원하고 있으므로 공격력이 가장 높은 장군개미의 공격력부터 차례대로 나눠주고 있습니다. 장군개미의 몫을 구하고, 남은 체력을 병정개미의 공격력으로 나누고, 또 남은 체력을 일개미의 공격력으로 나누는 것입니다.