프로그래머스 lvl 2. 더 맵게 [C++]

매운 것을 좋아하는 Leo는 모든 음식의 스코빌 지수를 K 이상으로 만들고 싶습니다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같이 특별한 방법으로 섞어 새로운 음식을 만듭니다.

섞은 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코비 지수 * 2)

 

Leo는 모든 음식의 스코빌 지수가 K 이상이 될 때까지 반복하여 섞습니다. Leo가 가진 음식의 스코빌 지수를 담은 배열 scoville과 원하는 스코빌 지수 K가 주어질 때, 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해서 섞어야 하는 최소 횟수를 return 하도록 solution 함수를 작성해주세요.

 

ex) scoville = [1, 2, 3, 9, 10, 12] and K = 7 일 때,

1. 스코빌 지수가 1인 음식과 2인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.
   새로운 음식의 스코빌 지수 = 1 + (2 * 2) = 5
   가진 음식의 스코빌 지수 = [5, 3, 9, 10, 12]
2. 스코빌 지수가 3인 음식과 5인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.
   새로운 음식의 스코빌 지수 = 3 + (5 * 2) = 13
   가진 음식의 스코빌 지수 = [13, 9, 10, 12]

모든 음식의 스코빌 지수가 7 이상이 되었고, 이때 섞은 횟수는 2회 입니다.

 

 

접근 point

• Priority Queue (오름차순 정렬)

 

Step 1. 오름차순으로 정렬되는 Priority Queue 생성하여 scoville 지수 값을 삽입한다.

 

 

Step 2. Priority Queue 맨 위의 값이 K보다 작거나 같을 때 까지 아래와 같이 반복한다.

 

 

Step 3. Priority Queue의 사이즈가 2보다 작고, 맨 위의 값이 K 보다 작을 때는 -1을 return 한다.

** 더 이상 스코빌 지수를 계산 할 음식 두 가지가 없는 것을 의미

 

 

#include <bits/stdc++.h>
using namespace std;

priority_queue<int, vector<int>, greater<int>> pq; // 오름차순

int solution(vector<int> scoville, int K) {
    int cnt = 0;
    for(int i = 0; i < scoville.size(); i++) {
        pq.push(scoville[i]);
    }
    
    while(true) {
        if(pq.size() >= 2) {
            if(pq.top() < K) {
                int a = pq.top(); pq.pop();
                int b = pq.top(); pq.pop();
                pq.push(a + 2 * b);
                cnt++;
            } else {
                break;
            }
        } else {
            if(!pq.empty()) {
                if(pq.top() < K) cnt = -1;
            }
            break;
        }
    }
    return cnt;
}