[99클럽] 16일차 문제: 더 맵게

 

 

오늘은 프로그래머스 더 맵게 문제가 출제됐다. 힙 -> 우선순위 큐를 사용해서 풀면된다. 월요일 특강에서도 나왔었는데 직접 배열로 힙 자료구조를 만들어 보면 좋을 것 같다. 언제 공부하지..?

 

import java.util.*;

class Solution {
    public int solution(int[] scoville, int K) {
        int answer = 0;
    
        PriorityQueue<Integer> q = new PriorityQueue<>();
        
        for(int i = 0; i < scoville.length; i++) {
            q.add(scoville[i]);
        }
        
        
        while(!q.isEmpty()) {
            int a = q.poll();
            
            if(a >= K) {
                return answer;
            }
            
            if(q.isEmpty()) {
                return -1;
            }
            
            int b = q.poll();
            
            int newNum = a + b * 2;
            answer++;
            q.add(newNum);
        }
        
        return -1;
    }
}