프로그래머스 L1 - 예산

2024. 3. 26. 14:49알고리즘/프로그래머스

https://school.programmers.co.kr/learn/courses/30/lessons/12982

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

그리디 알고리즘을 사용해서 해결했다.

가장 작은 금액을 신청한 부서부터 지원을 해주고 몇 개의 부서를 지원했는지 리턴하면 된다.

이 문제를 정리한 이유는 다른 사람의 풀이 중에 boolean 계산을 통해 간단하게 코드를 구현한 풀이가 있기에 정리한다.

function solution(d, budget) {
    return d.sort((a,b)=>a-b).reduce((count, price)=>{
        return count + ((budget-=price)>=0); 
    },0);
}

먼저 신청 금액을 오름차순 정렬을 한다. 그리고 reduce를 활용해 지원 갯수를 누적하는데 이때 boolean 연산이 들어간다.

예산금액에서 신청금액을 뺀 값이 0보다 크거나 같다면 count라는 정수와의 연산에 의해 1로 변환된다.

이렇게 되면 예산금액 범위에서 신청금액을 지원할 수 있을 때는 count에 1이 계속해서 누적된다.

만일 지원할 수 없는 경우라면 0이 누적된다. 그렇기에 지원 가능 부서의 갯수를 정확하게 구할 수 있다.

boolean 연산을 활용할 수 있는 재밌는 코드였다.