[백준 - 1927번] 최소 힙 - Java //Wello Horld//

이번에는 BOJ의 1927번 문제 "최소 힙"을 풀어보도록 하자

PriorityQueue를 이용해서 풀어주면 되는 아주 간단한 문제이다. 성공한 코드는 아래와 같다.

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

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

        int N = Integer.parseInt(br.readLine());
        PriorityQueue<Integer> queue = new PriorityQueue<>();
        for(int i = 0; i < N; i++){
            int a = Integer.parseInt(br.readLine());
            if(a == 0){
                if (queue.isEmpty()) {
                    bw.write("0\n");
                 } else {
                    bw.write(queue.poll() + "\n");
                 }
            } else {
                queue.add(a);
            }
        }

        bw.flush();
        br.close();
        bw.close();
    }
}

여기에서 한가지 집고 넘어가야 될 점은 PriorityQueue를 오름차순으로 만들어 줄려면 

PriorityQueue<Integer> queue = new PriorityQueue<>((a, b) -> b - a);

PriorityQueue의 Comperator를 작성할 부분에 위와같이 작성해주면 된다..

 

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

1927번: 최소 힙