[프로그래머스 / C++] 명예의 전당 (1)

문제

명예의 전당 (1) : 문제 링크

---

문제 분석

• "명예의 전당"이라는 TV 프로그램에서는 매일 1명의 가수가 노래를 부르고, 시청자들의 문자 투표수로 가수에게 점수를 부여한다. 매일 출연한 가수의 점수가 지금까지 출연 가수들의 점수 중 상위 k번째 이내이면 해당 가수의 점수를 명예의 전당이라는 목록에 올려 기념한다. 즉 프로그램 시작 이후 초기에 k일까지는 모든 출연 가수의 점수가 명예의 전당에 오르게 된다. k일 다음부터는 출연 가수의 점수가 기존의 명예의 전당 목록의 k번째 순위의 가수 점수보다 더 높으면, 출연 가수의 점수가 명예의 전당에 오르게 되고 기존의 k번째 순위의 점수는 명예의 전당에서 내려오게 된다.
• 이 프로그램에서는 매일 "명예의 전당"의 최하위 점수를 발표한다. 예를 들어, k = 3이고, 7일 동안 진행된 가수의 점수가 [10, 100, 20, 150, 1, 100, 200]이라면, 명예의 전당에서 발표된 점수는 아래와 같이 [10, 10, 10, 20, 20, 100, 100]이다.

일차	1	2	3	4	5	6	7
score	10	100	20	150	1	100	200
명예의 전당	10	100 10	100 20 10	150 100 20	150 100 20	150 100 100	200 150 100
발표 점수	10	10	10	20	20	100	100

• 명예의 전당 목록의 점수의 개수 k, 1일부터 마지막 날까지 출연한 가수들의 점수인 score가 주어졌을 때, 매일 발표된 명예의 전당의 최하위 점수를 return하는 solution 함수를 완성

• 제한 사항

• 3 ≤ k ≤ 100
• 7 ≤ score의 길이 ≤ 1,000
• 0 ≤ score[i] ≤ 2,000

• 바탕 컨테이너로 vector를 사용하는 오름차순의 우선순위 큐(priority_queue)를 사용할 것 이므로 queue 헤더를 포함
• 명예의 전당의 최하위 점수를 저장할 정수형 벡터 answer과 vector를 사용하여 오름차순의 우선순위 큐 pq를 초기화. for loop를 통해 score의 첫번째부터 마지막 원소까지 순환하고, if문을 통해 pq의 크기가 k보다 작다면(처음 명예의 전당에 k개의 원소를 채워야 하므로) score의 현재 인덱스 원소를 pq에 저장. else if문을 사용하여 현재 인덱스의 원소가 pq의 최소값보다 크다면, pq에서 그 최소값을 빼고 현재 인덱스의 원소를 저장. 이후, 명예의 전당의 최하위 점수에 해당하는 pq의 가장 작은 값을 answer에 저장하는것을 반복. loop 탈출 후, 최종적으로 저장된 answer을 return

우선순위 큐(priority_queue)
우선순위 큐는 선입선출 순이 아니라 우선순위가 높은 항목이 가장 앞에 오도록 하는 큐이다. 우선순위가 같은 항목들 사이에서도 선입선출 순을 보장하지 않고, 헤더에 함께 정의 되어 있고 정의는 다음과 같다.

template <typename _Ty, typename _Container = vector<_Ty>, typename _Pr = less<_Ty>>
class priority_queue;

두 번째 파라미터가 바탕 컨테이너이며 vector 외에도 deque을 사용할 수 있다. 세 번째 파라미터는 우선순위의 비교에 사용될 비교 연산을 나타내는 타입이다. less 템플릿 또는 greater 템플릿의 인스턴스가 오거나 클로져와 같은 function이 올 수 있다.

---

풀이

#include <vector>
#include <queue>

using namespace std;

vector<int> solution(int k, vector<int> score) {
    vector<int> answer;
    priority_queue<int, vector<int>, greater<int>> pq;
    
    for(int i = 0; i < score.size(); ++i) {
        if(pq.size() < k) pq.push(score[i]);
        else if(score[i] > pq.top()) {
            pq.pop();
            pq.push(score[i]);
        }
        answer.push_back(pq.top());
    }
    return answer;
}