[백준 - Java] 11286번 : 절댓값 힙

문제

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

설명

11279번 최대 힙, 1927번 최소 힙과 비슷한 문제로 정렬하는 부분만 바꿔주면 된다.

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
        int N = Integer.parseInt(br.readLine());
		
        // 절댓값이 같은 경우 그 중에서 작은 값으로 오름차순 정렬하고, 아닌 경우 절댓값이 작은 순서로 오름차순 정렬
        PriorityQueue<Integer> pq = new PriorityQueue<>((o1, o2) ->
            Math.abs(o1) == Math.abs(o2) ? Integer.compare(o1, o2) : Integer.compare(Math.abs(o1), Math.abs(o2))
        );
		
        for (int i = 0; i < N; i++) {
            int x = Integer.parseInt(br.readLine());
			
            if (x == 0) {
                if (!pq.isEmpty()) {
                    System.out.println(pq.poll());
                } else {
                    System.out.println(0);
                }
            } else {
                pq.add(x);
            }
        }

    }

}

GITHUB

https://github.com/MinchaeKwon/BOJ/blob/master/BOJ%2311286/src/Main.java