[프로그래머스] 더 맵게

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

프로그래머스

스코빌 지수가 기록된 scoville 배열에서 K 보다 작은 scoville 을 가진 원소가 있다면 최소 scoville 와, 그 다음으로 큰 scoville 을 이용하여 새로운 원소를 추가한다. 이를 위해서 최소힙을 사용한다.

#include <string>
#include <vector>
#include <queue>
#include <functional>
#include <iostream>

#define ll long long
using namespace std;

int solution(vector<int> scoville, int K) {
    priority_queue <int, vector<int>, greater<int> > pq;
    
    for(auto & ele : scoville)
    {
        pq.push(ele);
    }
    
    int cnt = 0; int mixed = 0; int first = 0; int second = 0;
    while(pq.size()!=1)
    {
        if(pq.top() >= K) break;
        first = pq.top();
        pq.pop();
        second = pq.top();
        pq.pop();
            
        mixed = first + second + second;
        pq.push(mixed);
        cnt++;
    }
    if(pq.top() < K) return -1;
    return cnt == 0 ? -1 : cnt;
}