[Lv.0] 개미 군단

https://school.programmers.co.kr/learn/courses/30/lessons/120837?language=javascript

거의 2달만에 문제를 풀려고 하니까 알던 함수도 생각이 안났다.

하루 한 문제라도 꾸준하게 풀어야겠다.

✅ 문제

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

제한 사항

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

입출력 예

hp	result
23	5
24	6
999	201

🔎 풀이 방법

주어진 hp를 나눌 때 5 > 3 > 1 순서로 나누며 내려간다.

5로 나눈 나머지를 변수에 저장해서 3으로 나누도록 내려주고, 3으로 나눈 나머지를 변수에 저장해서 1로 나눌 수 있도록 내려준다.

각 숫자로 나눈 몫을 합하면 최소 개미 수가 나온다.

hp가 24, 999인 경우, 원하는 result에 +1 되어 결과가 나왔다.

나누는 과정에서 몫이 소수점 단위로 나오고 있다는 것을 알고 parseInt()나 Math.floor()를 사용하면 될 것 같다고 생각했다.

둘 중 어떤 걸 사용할 지 고민하던 중 parseInt는 주로 문자열을 정수로 변환하는 경우, 진법을 변환하는 경우에 많이 사용되고, Math.floor는 실수에서 소수점을 버리고 내림을 하는데 많이 사용된다는 것을 알았다.

사용하고자 하는 목적에 부합하도록 Math.floor를 사용했다.

💻 코드

function solution(hp) {
    let answer = 0;
    if(hp / 5 !== 0) {
        answer += Math.floor(hp / 5);
        hp = hp % 5;
    }
    if(hp / 3 !== 0) {
        answer += Math.floor(hp / 3);
        hp = hp % 3;
    }
    if(hp !== 0) {
        answer += hp;
    }
    return answer;
}

👀 추가 공부

나머지 활용

function solution(hp) {
    return Math.floor(hp/5)+Math.floor((hp%5)/3)+(hp%5)%3;
}

 

깔끔한 코드

function solution(hp) {
    const 장군개미 = Math.floor(hp / 5);
    const 병정개미 = Math.floor((hp - (장군개미 * 5)) / 3);
    const 일개미 = hp - ((장군개미 * 5) + (병정개미 * 3));
    return 장군개미+병정개미+일개미;
}

 

~~ → Math.floor()

function solution(hp) {
    let ant = [5,3,1];
    let ans = [];

    for (let v of ant) {
        ans.push(~~(hp/v));
        hp%=v;
    }

    return ans.reduce((a,v)=>a+v,0);
}

 

forEach

function solution(hp) {
    let cnt = 0;
    [5,3,1].forEach(v => {
        cnt += Math.floor(hp/v);
        hp = hp % v;
    })
    return cnt;
}