#11286번 : 절대값 힙

문제

코드

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

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		
		PriorityQueue pq = new PriorityQueue<>(new Comparator() {
			@Override
			public int compare(Integer o1, Integer o2) {
				// o1 < o2 == -1 : 음수
				// o1 = o2 == 0 : 0
				// o1 > o2 == 1 : 양수
				Integer n = Math.abs(o1);
				Integer m = Math.abs(o2);
				
				if (n < m) {
					return -1;
				} else if (n > m) {
					return 1;
				} else {
					if (o1 < o2) {
						return -1;
					} else if (o1 > o2) {
						return 1;
					} else {
						return 0;
					}
				}
			}
		});
		
		
		StringBuilder sb = new StringBuilder();
		
		for (int i = 0; i < n; i++) {
			int x = Integer.parseInt(br.readLine());
			if (x == 0) {
				// 배열에서 가장 큰 값을 출력하고 그 값을 배열에서 제거
				if (pq.isEmpty()) {
					sb.append(0 + "\n");
				} else {
					sb.append(pq.poll() + "\n");
				}
			} else {
				// x가 자연수라면 배열에 x라는 값을 넣는(추가하는) 연산
				pq.offer(x);
			}
		}
		
		System.out.println(sb);
	}
}

출처

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