[백준11286] 절댓값 힙 Java

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

11286번: 절댓값 힙

 

package Chapter1;

import java.util.*;

public class BJ11286 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        PriorityQueue<Integer> q = new PriorityQueue<>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                if (Math.abs(o1) == Math.abs(o2)) {
                    return o1-o2;
                }
                else {
                    return Math.abs(o1) - Math.abs(o2);
                }
            }
        });

        for (int i=0; i<n; i++) {
            int query = sc.nextInt();
            if (query != 0) {
                q.offer(query);
            }
            if (query == 0) {
                if (q.isEmpty()) {
                    System.out.println(0);
                }
                else {
                    System.out.println(q.poll());
                }
            }
        }
    }
}