[프로그래머스] 더 맵게

---

🔗 문제 링크

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

프로그래머스

---

728x90

👩‍💻 코드

#include <string>
#include <vector>
#include <algorithm>
#include <functional>

using namespace std;

int solution(vector<int> scoville, int K) {
    int answer = 0;
    vector<int> heap = scoville;
    
    make_heap(heap.begin(), heap.end(), greater<int>());
    
    while(!heap.empty()){
        int min1 = heap.front();
        
        if(min1 >= K){
            return answer;
        }
        
        pop_heap(heap.begin(), heap.end(), greater<int>());
        heap.pop_back();
        
        if(heap.empty()){
            return min1 >= K ? answer : -1;
        }
        
        int min2 = heap.front();
        pop_heap(heap.begin(), heap.end(), greater<int>());
        heap.pop_back();
        
        heap.push_back(min1 + min2 * 2);
        push_heap(heap.begin(), heap.end(), greater<int>());
        
        answer++;
    }
    
    return answer;
}

---

📝 풀이

주어진 스코빌 지수 vector<int>로 최소 힙을 만듭니다.

최소값이 K이상이면 answer을 return 합니다.

최소값을 heap에서 pop하고 이 때 힙이 비어있고 최소값이 K 미만이면 -1을 return 합니다.

두 번째 최소값도 heap에서 pop하고 최소값과 두 번째 최소값을 섞어 새로운 음식을 만듭니다.

새로 만든 음식은 heap에 push하고 섞은 횟수(answer)을 증가시킵니다.

 

지금 보니 반복문의 조건을 잘못 설정한거 같습니다...😅