Java 프로그래머스 더 맵게

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

프로그래머스

문제 분류 : 코딩테스트 연습 > 힙(Heap) > 더 맵게

난이도 : 2

 

정답 코드 

import java.util.*;
class Solution {
    public int solution(int[] scoville, int K) {
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        for(int target : scoville){
            pq.add(target);
        }
        int cnt = 0;
        boolean flag = false;
        while(pq.size() >=1){
            if(pq.peek() >= K){
                flag = true;
                break;
            }
            cnt++;
            if(pq.size() == 1) break;
            int temp = pq.poll();
            int temp2 = pq.poll();
            pq.add(temp + temp2 * 2);
        }
        if(!flag) return -1;
        return cnt;
    }
}