본문 바로가기
Algorithm/Programmers

[Programmers] 예산(JavaScript)

by 백승전 2022. 3. 23.

 

알림

 

본 포스팅은 공부 목적으로 작성된 글이며 상업적 목적으로 절대 사용되지 않았음을 밝힙니다.

 

문제

 

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

 

코딩테스트 연습 - 예산

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

programmers.co.kr

 

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

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

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

 

풀이

 

신청금 d 배열을 일단 오름차순으로 정렬합니다. sort()를 사용하여 정렬하도록 합니다.(sort((a, b) => a - b) // 오름차순)

 

부서의 수(answer)와, 신청 금액(result)을 신청 금액이 budget보다 작으면 계속 더해주고 크면 직전 부서는 제외하고 이를 return 합니다.

 

코드(정확성 테스트 불통과)

 

function solution(d, budget) {
    
    let answer = 0; // 부서
    let sum = 0; // 금액
    
    d.sort((a, b) => a - b) // 오름차순 정렬
    
    for (let i = 0; i < d.length; i++) {
        if (sum < budget) { // 부서 금액 < 예산
            answer ++; // 부서 카운트
            sum += d[i]; // 금액 카운트
        } else {
            answer--; // 직전 부서 제외
        }
    }
    
    return answer;
}

 

정확성 문제를 통과하지 못했습니다.

 

출처: 프로그래머스

 

아무래도 코드를 훨씬 더 간단하고 빠르게(?) 작성이 가능할 것 같아 코드의 순서에 살짝 변화를 줘 봤습니다.

 

코드(정확성 테스트 통과)

 

function solution(d, budget) {
    
    let answer = 0; // 부서
    let sum = 0; // 금액
    
    d.sort((a, b) => a - b) // 오름차순 정렬
    
    for(let i = 0; i < d.length; i++){
        answer ++; // 부서 카운트
        sum += d[i]; // 금액 카운트
        
        // 금액이 예산보다 작으면 직전 부서 제외
        if(sum > budget) answer--;
    }
    
    return answer;
}

 

출처: 프로그래머스

 

처음부터 조건문으로 '금액이 예산보다 적으면 더하고, 초과되면 빼줘'가 아니라, 일단 계속 더하다가 '초과되면 빼줘'만 요청해서 코드도 훨씬 짧아졌고, 기능적으로도 긍정적인 모습을 보이는지 정확성 테스트에 통과하였습니다.

댓글