[Java] BOJ 1927 최소 힙

 

1927번: 최소 힙

---

정답 코드

• Java에서 힙으로 구현된 PriorityQueue 자료구조 활용
• PQ가 비었을때만 예외처리 필요

import java.io.*;
import java.util.*;

public class Main {

    static BufferedReader br;
    static int N;
    static PriorityQueue<Integer> pq;

    public static void main(String[] args) throws IOException {
        br = new BufferedReader(new InputStreamReader(System.in));
        pq = new PriorityQueue<>();
        N = Integer.parseInt(br.readLine());
        for (int i = 0; i < N; i++) {
            int input = Integer.parseInt(br.readLine());
            if (input == 0) {
                if (pq.isEmpty()) {
                    System.out.println(0);
                } else {
                    System.out.println(pq.poll());
                }
            } else {
                pq.add(input);
            }
        }
    }
}