[Java] 백준 1927 - 최소 힙

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

1927번: 최소 힙

min-heap을 구하는 간단한 문제이다.

다만 자바로 힙을 구하는 문제는 처음이다 보니 힙 자료구조가 구현된 클래스가 있는지 조차 잘 몰랐다.

그래서 구글링 해보니 PrioriryQueue라는 자료구조가 있었고 디폴트가 트리의 탑이 최솟값이 되는 min-heap이었기 때문에 클래스만 가져다 사용하면 문제해결이 가능했다.

import java.io.IOException;
import java.util.PriorityQueue;
import java.io.BufferedReader;
import java.io.InputStreamReader;

public class Main {

    public static void main(String[] args) throws IOException {
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int 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);
                    continue;
                }
                System.out.println(pq.poll());
                continue;
            }
            pq.offer(input);
        }
    }
}

https://docs.oracle.com/en/java/javase/17/docs/api/java.base/java/util/PriorityQueue.html

PriorityQueue (Java SE 17 & JDK 17)