[프로그래머스 / C++] 예산

문제

예산 : 문제 링크

문제 분석

• S사에서는 각 부서에 필요한 물품을 지원해 주기 위해 부서별로 물품을 구매하는데 필요한 금액을 조사했다. 그러나, 전체 예산이 정해져 있기 때문에 모든 부서의 물품을 구매해 줄 수는 없다. 그래서 최대한 많은 부서의 물품을 구매해 줄 수 있도록 하려고 한다. 물품을 구매해 줄 때는 각 부서가 신청한 금액만큼을 모두 지원해 줘야 한다. 예를 들어 1,000원을 신청한 부서에는 정확히 1,000원을 지원해야 하며, 1,000원보다 적은 금액을 지원해 줄 수는 없다. 부서별로 신청한 금액이 들어있는 배열 d와 예산 budget이 매개변수로 주어질 때, 최대 몇 개의 부서에 물품을 지원할 수 있는지 return 하도록 solution 함수를 완성
• 제한 사항

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

• 벡터의 오름차순 정렬을 위해 sort() 함수를 사용할 것 이므로 algorithm 헤더를 포함
• 최대 몇 개의 부서에 물품을 지원 할 수 있는지 저장할 정수형 변수 answer을 0으로 초기화. 최대한 많은 부서에 물품을 지원해야 하므로 신청 금액이 적은곳부터 지원해야 하므로 배열 d를 sort() 함수를 사용하여 오름차순으로 정렬. for loop를 통해 오름차순 정렬된 배열 d의 첫번째 원소부터 마지막까지 순환하고, if문을 사용하여 현재 budget보다 신청 금액이 크다면 loop를 탈출. 아니라면 budget에서 신청 금액을 빼고 저장하고, answer을 1씩 늘림. loop 탈출 후, 최종적으로 저장된 answer을 return

algorithm 헤더의 sort() 함수 사용법
void sort(T start, T end, Compare comp); //comp 인자가 공란이면 오름차순 정렬

• sort(v.begin(), v.end(), compare); // 사용자 정의 함수 사용
• sort(v.begin(), v.end(), greater<자료형>()); // 내림차순
• sort(v.begin(), v.end(), less<자료형>()); // 오름차순

풀이

#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<int> d, int budget) {
    int answer = 0;
    
    sort(d.begin(), d.end());
    for(int i = 0; i < d.size(); ++i) {
        if(budget < d[i]) break;
        budget -= d[i];
        answer++;
    }
    return answer;
}