카테고리 없음
프로그래머스>힙>더 맵게
연주우
2021. 4. 29. 20:58
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;
}
}