[Javascript] 프로그래머스 : 예산

문제

S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했습니다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없습니다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 합니다.

물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 합니다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없습니다.

 

부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성해주세요.

 

입출력 예시

d	budget	result
[1,3,2,5,4]	9	3
[2,2,3,3]	10	4

 

 

 

풀이

문제가 꽤 긴 편이지만 결국 최대 몇 개의 부서에 물품을 지원할 수 있는지를 묻고 있다.

신청한 금액이 적은 부서를 우선으로 해야 더 많은 부서에 지원할 수 있으니 일단 주어진 배열에 대해 내림차순 정렬이 필요하다.

그리고 budget에서 d의 요소를 하나씩 빼주면서 카운트를 하나씩 늘려가고 budget이 0이 되면 카운트를 리턴!

 

 

reduce로 풀고 싶었지만 생각처럼 잘 안되어서 처음에는 for문을 이용해서 풀었다.

 

for문으로 풀기

function solution(d, budget){
      d.sort((a,b) => a-b);

     for (let i = 0; i < d.length; i++) {
        budget -= d[i];
        answer += 1;

        if (budget < 0){
           answer--;
           break;  
        } 

    }

    return answer;
}

이후 다른 사람들의 풀이를 보고 reduce로 다시 풀었다.

 

 

 

reduce 메서드로 풀기

function solution(d, budget) {
    var answer = d.sort((a,b) => a - b).reduce((acc, curr) => acc + ((budget -= curr) >= 0), 0);
    return answer;
}

reduce 메서드를 이용해서 acc에 지원하기로 한 부서를 카운트한다.
0보다 작지 않을 때까지 budget에서 현재 값을 뺀다는 조건.

이번 기회에 reduce 함수를 몇 번 반복하다 보니 좀 더 익숙해진 것 같다.

 

 

reduce 함수에 대해서는 아래를 참고!

https://developer.mozilla.org/ko/docs/Web/JavaScript/Reference/Global_Objects/Array/Reduce

Array.prototype.reduce() - JavaScript | MDN