[백준] 11279 최대 힙

https://www.acmicpc.net/problem/11279

11279번: 최대 힙

 

이 문제는 PriorityQueue 라이브러리를 사용하면 해결할 수 있는 문제였다. 다만 최대 힙이기 때문에 Comparator를 사용해서 내림차순 기준으로 설정해주는 부분이 필요했다.

---

문제 및 입출력

 

---

코드

import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.Scanner;

public class Main
{
    public static void main(String[] args)
    {
        Scanner sc = new Scanner(System.in);
        StringBuilder sb = new StringBuilder();
        int n =sc.nextInt();
        
        // 우선순위 큐 설정(Comparator을 활용한 내림차순 설정)
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>(n, new Comparator<Integer>(){
           @Override
            public int compare(Integer x, Integer y)
            {
                return y-x;
            }
        });
        
        for(int i=0; i<n; i++)
        {
            int input = sc.nextInt();
            
            // 0을 입력받을 때 큐가 비었으면 0, 안 비었으면 최대값 출력
            if(input==0)
            {
                if(pq.isEmpty())
                {
                    sb.append(0 + "\n");
                }
                else
                {
                    sb.append(pq.poll() + "\n");
                }  
            }
            // 0이 아닌 값은 무조건 큐에 삽입
            else
            {
                pq.add(input);
            }
        }
        
        System.out.println(sb.toString());
        sc.close();
    }
}