[프로그래머스] 더 맵게

분류 : 힙

 

요구사항

최소 매움이 k보다 클때까지 조건대로 합차기

 

풀이

제일 안매운 거의 매움 정도 : heap의 탑 

2번째로 안매운거 매움 정도 : 1번째 pop한 뒤, 그 다음 top

두개를 조건에 따라 섞어 준뒤 다시 넣어 준다

 

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

using namespace std;

int solution(vector<int> scoville, int K) {
    int answer = 0;
    int flag=0;
    priority_queue<int> q;
    for(int i=0 ; i <scoville.size() ; i++){
        q.push(-scoville[i]);
    }
    while(!q.empty()){
        int now = -q.top();
     //   cout<<now<<endl;
        q.pop();
        if(now >=K){
           // answer++;
            flag=1;
            break;
        }
        if(!q.empty()){
            int next=-q.top();
         //   cout<<"B "<<next<<endl;
            q.pop();
            int sum = (now+next*2);
           // cout<<"sum"<<sum<<endl;
            answer++;
            q.push(-sum);
        }
        else{
            break;
        }
    }
    if(flag==1) return answer;
    else return -1;
}

int main(){
    return 0;
}