[C++/PGS] [PCCP 모의고사 #2] 2번 - 신입사원 교육

 

https://school.programmers.co.kr/learn/courses/15009/lessons/121688

프로그래머스

---

프로그래머스 PCCP 모의고사 2회 - 2번

소요시간 약 13분

자료구조, 추정 난이도는 level 1~2

 

범위가 커서 이중반복문이나 정렬 함수는 사용 못하는 문제!!

우선순위 큐 생각이 안나서 시간낭비했다 ㅠㅠ

 

나의 풀이

#include <string>
#include <vector>
#include <queue>
using namespace std;

int solution(vector<int> ability, int number) {
    int answer = 0;
    priority_queue<int, vector<int>, greater<int>> pq;
    
    for(int num: ability){
        pq.push(num);
    }
    
    for(int i=0; i<number; ++i){
        int sum = pq.top(); pq.pop();
        sum += pq.top(); pq.pop();
        pq.push(sum);
        pq.push(sum);
    }
    for(int num: ability){
        answer += pq.top();
        pq.pop();
    }
    return answer;
}