[프로그래머스 코딩테스트] 예산(Java)

  문제 설명

  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

import java.util.Arrays;
class Solution {
    public int solution(int[] d, int budget) {
        int answer = 0;
        Arrays.sort(d);
        int sum = 0;
        for (int i = 0; i < d.length; i++) {
            sum += d[i];
            if (sum <= budget) {
                answer++;
            } else {
                break;
            }
        }
        return answer;
    }
}

  우선 배열을 오름차순으로 정렬합니다. 최대한 많은 부서에 지원해야 하므로, 작은 수부터 먼저 취급하려는 것입니다. 지원하려는 금액을 담을 sum 변수를 생성했습니다.

  for문은 d의 길이만큼 순회합니다. 정렬된 각 원소의 값을 순서대로 sum에 더하기 할당 해줍니다. 만약 sum의 값이 예산보다 적거나 같다면 다시 for문을 순회할 것이고, 초과한다면 else의 break를 통해 for문을 빠져나올 것입니다. 값이 더해질 때마다 지원받는 부서가 하나씩 증가하는 것이므로, 마찬가지로 answer가 1씩 증가합니다.