[백준, 11286번] 절댓값 힙 구현하기 (JAVA)

 

우선순위 큐를 알아야 풀 수 있는 문제.

 

자료구조 오랜만에 보다보니까 까먹어서 우선순위 큐 재활치료하는데 시간이 좀 걸렸다....

어떤식으로 우선순위를 줄지 커스터마이징(?) 한 다음 for문 돌면서 조건에 맞게

출력할건 하고, 넣을건 넣고~~

public class 절댓값힙 {
    public static void main(String[] args) throws IOException {
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(bf.readLine());

        // 우선순위 큐 정렬 기준
        PriorityQueue<Integer> myQueue = new PriorityQueue<>((o1, o2) -> {
            int first_abs = Math.abs(o1);
            int second_abs = Math.abs(o2);

            if (first_abs == second_abs)  // 절댓값 같으면 작은 숫자에 우선순위를 줌.
                return o1 > o2 ? 1 : -1;
            else
                return first_abs - second_abs;
        });

        for (int i = 0; i < N; i++) {
            int request = Integer.parseInt(bf.readLine());
            if (request == 0) {
                if (myQueue.isEmpty())
                    System.out.println("0");
                else
                    System.out.println(myQueue.poll());  // 절댓값 가장 작은 수 출력 및 제거
            } else {
                myQueue.add(request);
            }
        }
    }
}