프로그래머스 더 맵게 (JAVA)

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

프로그래머스

 

저는 우선순위 큐를 통해 문제를 해결했습니다.

우선순위 큐를 통해 제일 낮은 값을 찾고 이 값이 K보다 작다면, 우선순위 사이즈를 보고 2이상이라면 두개를 꺼내서 새로운 스코빌 지수를 우선순위 큐에 넣었습니다.

이렇게 제일 낮은 값이 K보다 크게 되는지를 판단하였습니다.

 

import java.util.*;
class Solution {
    private static final int NOT_FOUND = -1;
    public int solution(int[] scoville, int K) {
        int answer = 0;
        PriorityQueue<Integer> p = new PriorityQueue<>();
        for(int num : scoville){
            p.offer(num);
        }
        while(true){
            int num = p.peek();
            if(num >= K){
                break;
            }else{
                answer++;
                if(p.size() <= 1){
                    answer = NOT_FOUND;
                    break;
                }
                int num1 = p.poll();
                int num2 = p.poll();
                p.offer(num1 + 2*num2);
            }
        }
        return answer;
    }
}