(Java) 백준 1927 - 최소 힙

PrioritiyQueue를 이용하면 간단하게 풀 수 있는 문제였다.

나는 If - eles 문으로 처리했지만

0을 고려해서 순서로 정렬해야 한다면

Compare 메서드를 오버라이딩 하면 될 것 같다.

PrioritiyQueue를 사용하지 않고 Heap을 직접 구현하는 분들도 계시던데

나중에 한번 시도해봐야겠다.

---

최종 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.PriorityQueue;

public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        PriorityQueue<Integer> pq = new PriorityQueue<>();
        int n = Integer.parseInt(br.readLine());

        for (int i = 0; i < n; i++) {
            int x = Integer.parseInt(br.readLine());
            if(x == 0) {
                if(!pq.isEmpty()) {
                    bw.write(pq.poll() +"\n");
                } else {
                    bw.write("0" + "\n");
                }
            } else {
                pq.offer(x);
            }
        }
        bw.flush();
        bw.close();
    }
}