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

문제 : programmers.co.kr/learn/courses/30/lessons/42626

코딩테스트 연습 - 더 맵게

주어진 scoville array를 priority queue에 넣기만 하면 쉽게 해결되는 문제

 

소스코드

import java.util.PriorityQueue;

class Solution {
    private int count = 0;

    public int solution(int[] scoville, int K) {
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        for (int factor : scoville) {
            pq.offer(factor);
        }

        while (pq.size() > 1) {
            int smallest = pq.poll();
            if (smallest >= K) {
                return this.count;
            }
            int nextSmallest = pq.poll();
            pq.offer(blend(smallest, nextSmallest));
        }

        return pq.poll() >= K ? this.count : -1;
    }

    private int blend(int a, int b) {
        this.count++;
        return a + 2 * b;
    }
}