11286번: 절댓값 힙 [JAVA]

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

11286번: 절댓값 힙

 

접근 방식)

1. "배열에서 절댓값이 가장 작은 값을 출력하고, 그 값을 배열에서 제거한다. 절댓값이 가장 작은 값이 여러개일 때는, 가장 작은 수를 출력하고, 그 값을 배열에서 제거한다." 도록 우선순위큐 정렬 정의

2. 조건 분기에 맞게 큐에 넣고, 빼고 출력하고 하도록 코드작성

 

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

public class B_11286 {
    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) -> {
            if (Math.abs(o1) == Math.abs(o2))
                return o1 - o2;
            return Math.abs(o1) - Math.abs(o2);
        });

        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < N; i++) {
            int input = Integer.parseInt(br.readLine());
            if (input == 0)
                sb.append(pq.isEmpty() ? 0 : pq.poll()).append('\n');
            else
                pq.add(input);
        }
        System.out.print(sb);
    }
}