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

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

11286번: 절댓값 힙

 

풀이

PriorityQueue를 사용하여 풀었다.

PriorityQueue를 선언할 때 조건에 맞는 Comparator 생성자를 사용하여 선언하면 된다.

코드

import java.util.*;
import java.io.*;

public class Main {

    public static void main(String[] args) throws IOException {
    	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    	BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
    	
    	PriorityQueue<Integer> pq = new PriorityQueue<>(new Comparator<Integer>() {
    		@Override
    		public int compare(Integer a, Integer b) {
    			if (Math.abs(a) == Math.abs(b)) {
    				return a - b;
    			}
    			return Math.abs(a) - Math.abs(b);
    		}
    	});
    	
    	int t = Integer.parseInt(br.readLine());
    	while (t-- > 0) {
    		int num = Integer.parseInt(br.readLine());
    		if (num != 0) {
    			pq.add(num);
    		} else {
    			if (pq.isEmpty()) {
    				bw.write(0 + "\n");
    			} else {
    				bw.write(pq.poll() + "\n");
    			}
    		}
    	}
    	
    	bw.flush();
    	br.close();
    	bw.close();
    }
}