[프로그래머스] 힙 : 더 맵게

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

코딩테스트 연습 - 더 맵게

풀이법:

1) 우선순위 큐를 사용해서 제일 작은 값을 K와 비교할 수 있게 한다.

2) 스코빌 지수 공식에 따라 계산한다.

 

Comment:

1) 예외처리하는게 좀 귀찮았다. 큐의 크기가 사이즈가 1 이하가 될때 peek() 또는 poll() 함수에서 런타임 에러 나는 부분을 확인 하는 등 짜잘한 예외상황을 생각해내는게 은근히 힘들다.

2) 실제 코테 상황에서 테스트 케이스들을 어떻게 확인하는지 알아보자

3) 답은 통과하는데 맞는 답인지는 잘 모르겠다 예외처리에 대해 덕지덕지 구현한 느낌이라

5) peek, poll, add, offer, remove 등 차이들을 정확히 알아보자

import java.util.*;

class Solution {
    public int solution(int[] scoville, int K) {
        int answer = 0;
        int count = 0;
        
        PriorityQueue pq = new PriorityQueue();
        
        for(int i = 0 ; i < scoville.length; i++){
            pq.add(scoville[i]);
        }
        
        while(!pq.isEmpty()){
            
            if(pq.size() <= 1 && (int)pq.peek() < K){
                return -1;
            }else if(pq.size() <= 1 && (int)pq.peek() >= K){
                break;
            }
            
            int temp1 = (int)pq.poll();
            int temp2 = (int)pq.peek();
            
            if(temp1 <= K){
                int new_sco = temp1 + (temp2 * 2);
                pq.poll();
                pq.add(new_sco);
                count++;
            }else{
                break;
            }
        }
        
        if(pq.isEmpty()){
            return -1;
        }else{
            return count;
        }
    }
}