[TIL] 2023.04.22 Programmers_예산

Today I Learned

• Programmers 문제 풀기
• 면접스터디

 

---

Programmers 문제 풀기

예산

문제 설명

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

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

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

제한사항

• d는 부서별로 신청한 금액이 들어있는 배열이며, 길이(전체 부서의 개수)는 1 이상 100 이하입니다.
• d의 각 원소는 부서별로 신청한 금액을 나타내며, 부서별 신청 금액은 1 이상 100,000 이하의 자연수입니다.
• budget은 예산을 나타내며, 1 이상 10,000,000 이하의 자연수입니다.

 

입출력 예

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

입출력 예 #1
각 부서에서 [1원, 3원, 2원, 5원, 4원]만큼의 금액을 신청했습니다. 만약에, 1원, 2원, 4원을 신청한 부서의 물품을 구매해 주면 예산 9원에서 7원이 소비되어 2원이 남습니다. 항상 정확히 신청한 금액만큼 지원해 줘야 하므로 남은 2원으로 나머지 부서를 지원해 주지 않습니다. 위 방법 외에 3개 부서를 지원해 줄 방법들은 다음과 같습니다.

• 1원, 2원, 3원을 신청한 부서의 물품을 구매해 주려면 6원이 필요합니다.
• 1원, 2원, 5원을 신청한 부서의 물품을 구매해 주려면 8원이 필요합니다.
• 1원, 3원, 4원을 신청한 부서의 물품을 구매해 주려면 8원이 필요합니다.
• 1원, 3원, 5원을 신청한 부서의 물품을 구매해 주려면 9원이 필요합니다.

3개 부서보다 더 많은 부서의 물품을 구매해 줄 수는 없으므로 최대 3개 부서의 물품을 구매해 줄 수 있습니다.

 

입출력 예 #2
모든 부서의 물품을 구매해 주면 10원이 됩니다. 따라서 최대 4개 부서의 물품을 구매해 줄 수 있습니다.

 

 

나의 1차 풀이

function solution(d, budget) {
    let sum = 0;
    d.sort((a,b) => a - b);
    for (const index in d) {
        sum += d[index];
        if (sum === budget) return parseInt(index) + 1;
        if (sum > budget) return parseInt(index);
    }
}

배열을 sort 한 후에 for문을 돌면서 sum을 더해가는 방식으로 풀어봤는데 3, 6, 18, 19번 테스트 케이스를 통과하지 못했다.

 

 

나의 2차 풀이

function solution(d, budget) {
    let sum = 0;
    d.sort((a,b) => a - b);
    if (d.length === 1) return 1;
    for (const index in d) {
        sum += d[index];
        if (sum === budget) return parseInt(index) + 1;
        if (sum > budget) return parseInt(index);
        if (parseInt(index) === d.length - 1) return parseInt(index) + 1;
    }
}

예외 처리가 부족하다고 생각해서 if문을 추가하다 보니 지저분한 풀이가 되었다. 하지만 여전히 테스트 테이스 3번 하나를 통과하지 못했다.

 

 

나의 3차 풀이

function solution(d, budget) {
    d.sort((a,b) => a - b);
    if (d.length === 1) return 1;
    for (const index in d) {
        budget -= d[index];
        if (budget === 0) return parseInt(index) + 1;
        if (budget < 0) return parseInt(index);
        if (parseInt(index) === d.length - 1) return parseInt(index) + 1;
    }
}

sum을 0부터 시작해서 더하는 방식 말고 budget부터 시작해서 0이 될 때까지 숫자를 빼는 방식은 어떨까 싶어서 새로 풀어봤지만 이 풀이에서도 if문이 너무 많았고 결과적으로 문제를 통과하지 못했다.

 

 

나의 최종 풀이

function solution(d, budget) {
    let sum = 0;
    let answer = 0;
    d.sort((a,b) => a - b);
    for (const num of d) {
        sum += num;
        if (sum > budget) break;
        answer++;
    }
    return answer;
}

for문 안에서 answer에 +1을 하기 전에 if문 조건에 따라 break 하고, for문 밖에서 answer를 return 해야 문제를 통과할 수 있었다.

1. d 배열을 오름차순으로 sort 한다.
2. for문을 돌면서 sort 한 배열의 각 요소를 sum에 더해준 후 if문을 통과한다면 answer도 +1을 한다.
3. 만약 sum이 budget보다 커진다면 for문을 중단하고 answer를 return 한다.

 

다른 사람의 풀이

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

배열을 sort 한 후에 while문 조건이 true라면 (배열의 모든 요소의 합이 budget보다 크다면) 배열의 마지막 요소를 하나씩 제거한다. while문을 돌면서 배열의 마지막 요소를 차례대로 제거하고, 배열의 모든 요소의 합이 budget보다 크지 않은 경우가 발생하면 배열의 length를 반환한다. 재미있는 답변이긴 하지만 reduce 메서드를 너무 많이 사용해서 효율적인 답변인지는 잘 모르겠다.

 

---

회고

면접스터디 진행할 때마다 아직 한참 부족하다는 것을 느낀다. 더 열심히 노력해야겠다 🥹🥹