[Java / C++] 예산

문제

https://programmers.co.kr/learn/courses/30/lessons/12982

코딩테스트 연습 - 예산

 

 

코드

Java

import java.util.Arrays;

class Solution {
    public int solution(int[] d, int budget) {
        int answer = 0;
        
        Arrays.sort(d);
        
        int i=0;
        while( i<d.length && budget >= d[i]){
            answer++;
            budget -= d[i++];
        }
        
        return answer;
    }
}

 

 

C++

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

using namespace std;

int solution(vector<int> d, int budget) {
    int sum=0;
    int temp = budget;
    sort(d.begin(), d.end());
    for(int i=0; i<d.size(); i++)
        printf("%d", d[i]);

    for (int i=0; i<d.size(); i++){
        budget=budget-d[i];
        sum=sum+d[i];
        printf("\n%d", sum);
        if(sum > temp)
            return i;
    }

}