[Java][프로그래머스][Level 1] 예산

문제 설명

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

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

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

 

제한사항

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

생각

어떻게 하면 최대한 많은 부서에 예산을 지원해 줄 수 있는지 생각해보자. 예를 들어, 각 부서의 예산이 [10,4,6,2,3,2] 이고 예산은 15라고 해보자. 앞에서부터 10의 예산을 지원해주면 5가 남는다. 후에 4의 예산을 지원해주면 1이 남는다. 1로는 더 이상 예산을 지원해 줄 수 없으므로 지원해준 부서는 2개가 된다.

 

가장 많이 지원해 주려면 가장 예산이 적은 부서부터 지원해주면 된다. 나는 이를 위해 먼저 오름차순으로 정렬을 하고 생각하고자 하였다. 오름차순을 하면 [2,2,3,4,6,10]이 된다. 그에 맞게 예산은 13,11,8,4가 되며 4개의 부서까지 지원을 해준다. 6의 예산을 필요로 하는 부서에게는 지원을 해줄 수 없다. 이 경우를 항상 검사해주면서 문제를 풀어가면 될 것이다. 

 

매 부서마다 예산이 0이 되었는지, 혹은 남은 예산이 부서에서 원하는 돈보다 적은지를 확인하여 이 둘 중 하나라도 만족하면 더 이상 나눠줄 수 있는 예산이 없는 것이다. 오히려 말로 설명하니 복잡한 것 같아 바로 코드를 보며 얘기하겠다.

 

코드

import java.util.*;
 
class Solution {
      public int solution(int[] d, int budget) {
          //answer는 예산을 지원해준 부서의 개수이다.
          int answer = 0;
          Arrays.sort(d); //오름차순
          
 
          for(int i=0; i<d.length; i++) { 
            //예산이 0이 되거나 지원해주려는 부서보다 작으면 종료한다.
            if(budget == 0 || budget < d[i]) {
                break;
            }
            budget -= d[i];
            answer++;
          }
          return answer;
      }
    }

문제의 설명이 길어 복잡해보이지만, 곰곰이 생각하면 금방 풀 수 있는 문제였다.