프로그래머스 - 더 맵게

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

코딩테스트 연습 - 더 맵게

 

 

 

 

 

우선순위큐가 힙을 이용해서 구현을 한다고 한다. 원소를 추가할 때마다 값을 정렬을 해준다.

이 문제는 우선순위큐를 써야 효율성 테스트를 통과할 수 있다고 한다.

 

 

 

 

 

코드

import java.util.*;

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

 

 

 

 

 

 

참고

PriorityQueue(우선순위큐) : https://coding-factory.tistory.com/603

[Java] PriorityQueue(우선순위 큐) 클래스 사용법 & 예제 총정리

 

풀이 : https://jar100.tistory.com/18

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