[프로그래머스] 고득점 Kit 힙(Heap)

42626 더 맵게

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

코딩테스트 연습 - 더 맵게

 

20220316 C++ 문제 풀이

#include <string>
#include <vector>
#include <queue>
#include <algorithm>

using namespace std;

int solution(vector<int> scoville, int K) {
    int answer = 0;
    priority_queue <int, vector<int>, greater<int> > pq;
    
    for(int i=0; i<scoville.size(); i++) {
        pq.push(scoville[i]);
    }
    while(pq.top() < K) {
        if(pq.size() <= 1) return -1;
        int a = pq.top();
        pq.pop();
        int b = pq.top();
        pq.pop();
        
        pq.push(a + 2*b);
        answer++;
    }
    
    return answer;
}

우선순위 큐 사용 (Heap 구조)

include <queue>

priority_queue<int> pq;    // 선언 큰수가 top