d | budget | result |
---|---|---|
[1,3,2,5,4] | 9 | 3 |
[2,2,3,3] | 10 | 4 |
각 부서에서 [1원, 3원, 2원, 5원, 4원]만큼의 금액을 신청했습니다. 만약에, 1원, 2원, 4원을 신청한 부서의 물품을 구매해주면 예산 9원에서 7원이 소비되어 2원이 남습니다. 항상 정확히 신청한 금액만큼 지원해 줘야 하므로 남은 2원으로 나머지 부서를 지원해 주지 않습니다. 위 방법 외에 3개 부서를 지원해 줄 방법들은 다음과 같습니다.
3개 부서보다 더 많은 부서의 물품을 구매해 줄 수는 없으므로 최대 3개 부서의 물품을 구매해 줄 수 있습니다.
모든 부서의 물품을 구매해주면 10원이 됩니다. 따라서 최대 4개 부서의 물품을 구매해 줄 수 있습니다.
c++
#include <iostream>
#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(d[i] > budget) break;
budget -= d[i];
answer++;
}
return answer;
}
budget
내에서 최대한 많은 부서를 지원하려면 최대한 작은 금액 여러 개를 가져가야 함vector
를 오름차순 정렬, for
문으로 최소 금액부터 차례로 budget
에서 차감d[i]
, 즉 i
번째 vector element
인 금액
이 남아 있는 budget
보다 클 경우 breakanswer
의 카운트값 증가💡
sort
함수 :algorithm
라이브러리 내에 있으며 정렬할 때 사용
sort(v.rbegin(), v.rend())
와 같이 사용하여 내림차순 정렬도 표현 가능함
💡size()
함수 :vector
의element
개수를 알 수 있음
처음 답안 작성할 때 생각한 방법은 위에 있는 정렬을 사용하는 것과 함수 min_element
를 사용하는 것 두 가지가 있었다
정렬하지 않고 함수로 최솟값
과 최솟값의 인덱스
를 구하고, 마찬가지로 남은 예산을 넘지 않으면 예산에서 차감하고 카운트값 증가 후, erase
함수로 vector
내에서 해당 element
를 삭제하는 대충 그런 방식...
int solution(vector<int> d, int budget) {
int answer = 0, min = 0, min_elem = 0;
do {
min = *min_element(d.begin(), d.end());
min_elem = min_element(d.begin(), d.end()) - d.begin();
if (min <= budget) {
budget -= min;
answer++;
d.erase(d.begin() + min_elem);
}
}
while (budget >= min);
return answer;
}
뭔가 될 것도 같았는데 테스트 케이스는 둘 다 통과했고 실채점에서도 86점 얼마가 나오긴 했다 근데 몇 가지 케이스에서 아무리 수정해도 segmentation fault (core dumped)가 오지게 뜨더라
💡 다른 풀이들도 다 훑어 봤는데 일단
sort
쓰는 게 가장 베스트인 것 같고 맞는 방법 같다 정렬 이후로 어떻게 하느냐에 따라 더 간결한 표현이 가능하겠지만