[BOJ 1927] 최소 힙 (Java)

문제

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

1927번: 최소 힙

 

풀이

PriorityQueue로 풀면 된다. 

PriorityQueue는 정수형의 경우 기본정렬이 오름차순이므로 따로 Comparator를 만들어줄 필요는 없다.

 

소스코드

package heap;

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

public class BOJ1927 {

    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> pque = new PriorityQueue<Integer>();

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