[프로그래머스][Level2] 더 맵게 c++

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

프로그래머스

 

풀이과정

우선순위 큐는 기본 정렬이 내림차순이기 때문에 greater<int>를 통해 오름차순으로 정렬해준다.  
우선순위 큐에 스코빌 지수를 넣고, 무한 반복문을 돌면 되는데 탈출 조건이 두가지가 있다.

첫번째로는 pq.size()가 2보다 작으면 pop을 두번 할 수 없고  
모든 음식의 스코빌 지수를 K이상으로 만들 수 없다는 뜻이기 때문에 -1을 return 해준다.

두번째로는 pq.top()이 K보다 크거나 같을 때이다.  
이땐 가진 스코빌의 지수가 다 K 이상이 되었다는 뜻이므로 카운트한 answer을 return 해준다.

 

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

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(1){
        if(pq.size() < 2) return -1; // 탈출 조건 1

        int first = pq.top();
        pq.pop();
        int second = pq.top();
        pq.pop();
        pq.push(first + second * 2);
        answer++;

        if(pq.top() >= K) return answer; // 탈출 조건 2
    }
}