프로그래머스(Level 2) - 더 맵게

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

프로그래머스

heap과 관련한 문제를 풀어보았습니다. 개념에 대한 공부를 먼저 하고 우선순위 큐라는 자료구조를 미리 알았기 때문에 쉽게 풀 수 있었습니다.

import java.util.*;

class Solution {
    public int solution(int[] scoville, int K) {
        int answer = 0;
        int temp = 0;
        PriorityQueue<Integer> heap = new PriorityQueue<>();
        
        for(int i: scoville){
            heap.offer(i);
        }
        
        while(true){
            
            if(heap.peek() >= K)
                break;
            
            if(heap.peek() == 0 || heap.size() == 1){
                return -1;
            }
            
            int min = heap.poll();
            int second = heap.poll();
            
            temp = min+ 2*second;
            
            heap.offer(temp);
            answer++;
        }
        return answer;
        
    }
}