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

[프로그래머스] 더 맵게

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

문제접근

주어진 음식 배열에서 스코빌지수가 가장 낮은 두 음식을 섞어서 기준인 K를 넘겨야 한다. 두 음식을 섞은 결과도 다시 음식 배열에 추가해서 연산을 해야하기 때문에, 계속해서 오름차순을 유지할 수 있는 최소힙을 사용했다.

Code

import java.util.PriorityQueue;

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

배운 점 / 어려웠던 점 / 느낀 점

PriorityQueue를 사용해서 최소 힙을 간단하게 사용할 수 있다.
선언할 때 new PriorityQueue<>(Collections.reverseOrder()); 를 사용하면 최대 힙으로 사용할 수 있다.

// 최소 힙
PriorityQueue<Integer> minHeap = new PriorityQueue<>();

// 최대 힙
PriorityQueue<Integer> maxHeap = new PriorityQueue<>(Collections.reverseOrder());