[프로그래머스] 개미 군단

문제

문제 설명

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

풀이

방식

• 각 개미 공격력에 대한 배열을 만든다.
• 공격력 배열에 대한 향상된 for문을 사용한다
• hp에서 개미 공격력을 나눈 몫을 answer에 더한다.
• hp에서 개미 공격력을 나눈 나머지를 hp로 바꾼다.

제출 코드

class Solution {
    public int solution(int hp) {
        int[] ants = {5, 3, 1};
        int answer = 0;
        for (int ant : ants) {
            answer += hp / ant;
            hp = hp % ant;
        }
        return answer;
    }
}

다른 사람의 풀이

코드

class Solution {
    public int solution(int hp) {
        return hp / 5 + (hp % 5 / 3) + hp % 5 % 3;
    }
}

• hp를 공격력으로 나눈 나머지와 몫을 더해주었다.
• 코딩테스트라 개미가 추가될 일이 없다면 이렇게 사용하면 좋고 나중에 개미가 추가된다면 내 코드가 더 낫지 않을까 싶다.