[프로그래머스] 더 맵게 - Java

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

프로그래머스

 

프로그래머스 더 맵게 자바 풀이

난이도: Lv2

 

풀이

우선순위 큐를 사용한다.

큐에서 뽑은 것이 K보다 작은 경우 가장 작은 스코빌지수가 아직 K보다 작으므로,

가장 작은 스코빌 지수를 poll()하고 두번째 poll()한것을 두배를 하여 더한다.

이렇게 반복을 하며 K이상일 경우를 구한다.

 

 

 

코드

import java.util.PriorityQueue;

class Solution {
    public int solution(int[] scoville, int K) {
        int answer = 0;
        PriorityQueue<Integer> queue = new PriorityQueue<>();

        for(int i=0;i<scoville.length;i++){
            queue.add(scoville[i]);
        }

        while (queue.peek()<K) {
            if(queue.size()==1){
                return -1;
            }
            queue.add(queue.poll()+queue.poll()*2);
            answer++;
        }
        return answer;
    }
}