백준
백준[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의 정렬조건에 그냥 만들면 된다는 것을 알고 쉽게 해결했다.