자바(백준) 11286 절댓값 힙

오답 노트 & 새로 알게 된 점

해당 문제는 우선순위 큐를 만들 때, comperator를 잘 구현할 수 있는지 물어보는 문제였다.

 

---

코드

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.PriorityQueue;
 
public class Main {
    static int atoi(String str) {
        return Integer.parseInt(str);
    }
 
    static int N;
    static PriorityQueue<Integer> pq = new PriorityQueue<>((o1,o2)->{
        if(Math.abs(o1) == Math.abs(o2))
            return o1 - o2;
        else
            return Math.abs(o1) - Math.abs(o2);
    });
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
 
        N = atoi(br.readLine());
 
        for (int i = 0; i < N; i++) {
            int cmd = atoi(br.readLine());
 
            if(cmd == 0){
                if(pq.isEmpty()) sb.append(0).append("\n");
                else sb.append(pq.poll()).append("\n");
            } else pq.offer(cmd);
        }
 
        System.out.print(sb);
    }
}