[힙] 더 맵게 - JAVA

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

코딩테스트 연습 - 더 맵게

---

반복문을 통해서 최솟값 2개를 반복해서 뽑아내면 되니까 우선순위 큐를 사용하여 구현했다

별건없다!

 

자바 코드)

import java.util.*;

class Solution {
    public int solution(int[] scoville, int K) {
        int answer = 0;
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        for(int s : scoville) {
            pq.add(s);
        }
        while(pq.peek()<K) {
             if(pq.size()==1) {
                return -1;
            }
            int min1 = pq.poll();
            int min2 = pq.poll();
            pq.add(min1+min2*2);
            answer++;
        }
        return answer;
    }
}