BOJ - 최대 힙

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

11279번: 최대 힙

• 최대 힙, 최소 힙은 우선수위큐를 이용하면 풀수 있다.
• Number Class를 정의하고 내림차순으로 변경하였다.

 

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


public class Main {
    public static final BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

    public static void main(String[] args) throws Exception {
        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());

        PriorityQueue<Number> queue = new PriorityQueue<>();

        while(N-- > 0) {
            Integer number = Integer.parseInt(br.readLine());
            if(number == 0) {
                if(queue.isEmpty()) {
                    System.out.println("0");
                } else {
                    System.out.println(queue.poll().number);
                }
            } else {
                queue.offer(new Number(number));
            }
        }
    }

    public static class Number implements Comparable<Number> {
        private Integer number;

        public Number(Integer number) {
            this.number = number;
        }

        @Override
        public int compareTo(Number other) {
            return other.number - this.number;
        }
    }
}