[프로그래머스/JAVA] 더 맵게

https://programmers.co.kr/learn/courses/30/lessons/42626?language=java 

코딩테스트 연습 - 더 맵게

import java.util.*;

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

힙(우선순위 큐)은 자동 오름차순 정렬이 된다는 것을 이용해 제일 작은 값과 하나 더 큰 값을 꺼내 연산 후 비교한다.

 

- 참고

https://jar100.tistory.com/18

프로그래머스 알고리즘 : 더 맵게 (java)

https://hidelookit.tistory.com/41

[프로그래머스] 힙(Heap) - 더 맵게 (Java)