[백준 11286] 절댓값 힙

by

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

11286번: 절대값 클러스터

작업 수 N(1≤N≤100,000)은 첫 번째 줄에 지정됩니다.

다음 N 줄은 작업에 대한 정보를 나타내는 정수 x를 받습니다.

x가 0이 아니면 x 값을 배열에 삽입(더하기)하는 연산이고 x는 0

'추천 관련글,

www.acmicpc.net

우선순위 큐로 해결할 수 있었습니다.

음수이면 q2에 배치되고 q2는 내림차순으로 정렬된 대기열입니다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Collections;
import java.util.PriorityQueue;

public class Main {

	public static void main(String() args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		PriorityQueue<Integer> q = new PriorityQueue<>();
		PriorityQueue<Integer> q2 = new PriorityQueue<>(Collections.reverseOrder());
		for (int i = 0; i < n; i++) {
			int num = Integer.parseInt(br.readLine());
			if (num == 0) {
				if (q.isEmpty()) {
					System.out.println(0);
					continue;
				}
				if (!
q2.isEmpty()) {
					if (q.peek() == Math.abs(q2.peek())) {
						q.poll();
						System.out.println(q2.poll());
					} else
						System.out.println(q.poll());
				} else
					System.out.println(q.poll());
			} else {
				q.add(Math.abs(num));
				if (num < 0)
					q2.add(num);
			}
		}
	}

}