본문 바로가기

카테고리 없음

프로그래머스>힙>더 맵게

import java.util.PriorityQueue;

//우선순위 큐에서 peek 는 고르되, 제거는 안 함
//poll 은 아예 고른 후 제거

class Solution 
{
    public int solution(int[] scoville, int K) 
    {
        int answer = 0;
        
        PriorityQueue<Integer> qu = new PriorityQueue<>();
        
        for(int i : scoville)
            qu.offer(i);
        
     
        int cnt=0;
        
        for(int i=0; i<scoville.length; i++)
        {
            if(qu.peek()< K)
            {
                qu.add(qu.poll() + qu.poll()*2);
                
                cnt ++;
                //System.out.println(cnt);
                //System.out.println(qu);
            }
            
        }
               
               answer =cnt; 
        
          
            
     
            
        return answer;
    }
}