Coding 01

백준[11286]번 : 절댓값 힙 ( JAVA ) 본문

백준

백준[11286]번 : 절댓값 힙 ( JAVA )

하루우울루 2023. 7. 25. 13:06

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

 

11286번: 절댓값 힙

첫째 줄에 연산의 개수 N(1≤N≤100,000)이 주어진다. 다음 N개의 줄에는 연산에 대한 정보를 나타내는 정수 x가 주어진다. 만약 x가 0이 아니라면 배열에 x라는 값을 넣는(추가하는) 연산이고, x가 0

www.acmicpc.net


Code

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.PriorityQueue;

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));
		
		int t =Integer.parseInt(br.readLine());
		
		PriorityQueue<Integer> queue = new PriorityQueue<Integer>((a,b) -> {
			if(Math.abs(a) - Math.abs(b) == 0) {
				return a-b;
			}
			else {
				return Math.abs(a) - Math.abs(b);
			}
		}); // 정렬의 조건을 정해주었다
		
		while(t-- > 0) {
			int n = Integer.parseInt(br.readLine());
			
			if(n != 0) {
				queue.add(n);
			}
			else {
				if(queue.isEmpty()) {
					bw.write(0+"\n");
				}
				else {
					bw.write(queue.poll()+"\n");
				}
			}
		}
		
		br.close();
		bw.close();
	}
	
}

절댓값의 조건을 어떻게 처리해야 하나 생각하다가 PriorityQueue의 정렬조건에 그냥 만들면 된다는 것을 알고 쉽게 해결했다.