N

(프로그래머스 c++)예산 본문

프로그래머스 알고리즘/1단계

(프로그래머스 c++)예산

naeunchan 2020. 4. 29. 11:21
728x90
반응형

sort를 이용하여 푸는 문제..!

우선 <algorithm>을 include 하여 sort를 쓰도록 하자..!

sort는 기본적으로 오름차순 정렬이기 때문에 sort(d.begin(), d.end());만 써주면 된다..!

그리고 int형 변수 sum을 선언한다.

for문을 통해 sum + d [i]가 budget보다 작으면 더하여 answer++을 해주고,

더한 값이 budget보다 크다면 바로 answer를 리턴한다..!

easy..!

#include <iostream>
#include <stdio.h>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<int> d, int budget) {
    int answer = 0, sum = 0;
    
    sort(d.begin(), d.end());
    
    for(int i = 0; i < d.size(); i++)
    {
        if(sum + d[i] <= budget)
        {
            sum += d[i];
            answer++;
        }
        else
            return answer;   
    }
}
728x90
반응형