[프로그래머스] 예산

프로그래머스의 Lv.1 예산 문제 풀이

문제 설명

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

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

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

제한 사항

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

문제 풀이
이 문제에서의 조건은 부서가 신청한 금액을 정확히 지원을 해줘야한다는 것으로 만약 지금까지 지원한 금액과 다음 지원할 부서의 신청금의 합이 예산을 넘어가면 다음 지원할 부서는 지원을 못해주는 것으로 조건문을 넣어서 문제를 풀었습니다. 순회를 하면 조건문을 만족시키면 멈추고, 조건문이 충족되지 않았다면 다시 순회하면서 지원금을 계산하고 몇개의 부서를 지원했는지 계산을 하였습니다.

코드

public int solution(int[] d, int budget) {
   int answer = 0;
   int count = 0;

   Arrays.sort(d);
   for (int i = 0; i < d.length; i++) {
     if (answer + d[i] > budget) {
       break;
     }
     answer += d[i];
     count++;
   }
   return count;
 }