[BOJ 1927] 최소 힙(java)

문제

최소 힙

문제 풀이

자료 구조중 하나인 힙(heap)에 관한 문제이다. 이클립스 내에 구현된 우선순위 큐를 import하여 문제를 풀었다.

프로그램 코드

public class BOJ1927_최소힙 {
	static int N;
	static PriorityQueue<Integer> pq = new PriorityQueue<>();

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

		for (int i = 0; i < N; i++) {
			int input = Integer.parseInt(br.readLine());
			if (input == 0) {
				if (pq.isEmpty()) { // 큐가 비어있으면 0 출력
					bw.write(0 + "\n");
				} else { // 큐가 비어있지 않으면 삭제&출력
					bw.write(String.valueOf(pq.poll()) + "\n");
				}

			} else { // 0이 아닌 다른 숫자 삽입
				pq.offer(input);
			}
		}
		bw.flush();
		bw.close();
		br.close();
	}

}