[CodingTest] 개미 군단

📖 Exam

문제 설명

개미 군단이 사냥을 나가려고 합니다. 개미군단은 사냥감의 체력에 딱 맞는 병력을 데리고 나가려고 합니다. 장군개미는 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합니다.

---

✍ Answer

class Solution {
    public int solution(int hp) {
        int answer = 0;
        int count = 0;
        int ant1 = 5;
        int ant2 = 3;
        int ant3 = 1;

        if(hp==1) {
            return 1;
        } else if(hp==2){
            return 2;
        } else if(hp==3){
            return 1;
        } else if(hp==4){
            return 2;
        } else if(hp==5){
            return 1;
        } else if(hp%ant1 == 0){
            return hp/ant1;
        } else {
            count = hp / ant1;
            answer = hp % ant1;
            if (answer % ant2 == 0) {
                return count + answer / ant2;
            } else if(hp % ant1 < ant2){
                return count + answer / ant3;
            } else {
                return count + answer / ant2 + (answer%ant2) / ant3;
            }
        }
    }
}

---

💡 Realization

• 너무 멍청하게 풀었는데 당장에 저거 말고 생각이 안남. ㅠ

• 아래처럼 푸는거 생각했는데 while 문 할께 아니라 그냥 순차 진행하면 되는거였네.ㅠㅠ

class Solution {
    public int solution(int hp) {
        int answer = hp / 5;
        hp %= 5;

        answer += hp / 3;
        hp %= 3;

        answer += hp / 1;

        return answer;
    }
}

참고사이트: https://school.programmers.co.kr/learn/courses/30/lessons/120837/solution_groups?language=java