[백준] 1927 최소 힙 (JAVA)

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

1927번: 최소 힙

풀이

이 문제는 힙을 이용해 구현하면 되는 문제로 나는 자바를 이용하기 때문 힙을 이용해 구현하였다. 풀이 방법은 문제에서 시킨대로 맨 처음 입력받느 N만큼 입력을 받고 이 때 만약 입력받은 값이 0이라면 힙에서 우선순위가 가장 높은 수를 뽑아서 출력하면 된다. 만약 힙이 비어있다면 0을 출력한다. 그리고 0이 아닌 임의의 수가 입력된다면 힙에 추가한다.

 

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

class Main{

    public static void main(String[] args) throws IOException {
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int n = Integer.parseInt(br.readLine());

        for (int i = 0; i < n; i++) {
            int temp = Integer.parseInt(br.readLine());

            if(temp == 0){
                if(pq.isEmpty())
                    bw.write(0 + "\n");
                else {
                    bw.write(pq.poll() + "\n");
                }
            }

            else {
                pq.add(temp);
            }
        }

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

    }
}​