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

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

프로그래머스

 

힙 연습문제 답게 우선순위 큐를 사용하면 엄청 쉽게 풀린다.

import java.util.*;

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

 

테스트 케이스 3개정도가 통과 안돼서 다시 봤더니 모든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1을 리턴해야 된다. 문제를 꼼꼼히 읽자..