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

🔗 문제 링크

BOJ 11286번: 절댓값 힙

11286번: 절댓값 힙

 

📝 풀이 과정

[백준] 11279번: 최대 힙과 정렬부분만 다른 문제이다. 

 

 

우선 순위는 절대값이 작은 순 → 숫자가 작은 순으로 제거해야 하기 때문에 PriorityQueue를 사용해 Comparator의 compare를 오버라이딩하여 Queue에서 올바른 순서로 나올 수 있게 만들어 주었다.

 

PriorityQueue<Integer> que = new PriorityQueue<>((o1, o2) ->
    Math.abs(o1) == Math.abs(o2) ? Integer.compare(o1, o2) : Integer.compare(Math.abs(o1), Math.abs(o2))
);

삼항 연산자를 통해 작성하였는데 조건? true일 경우 : false일 경우의 형태를 가지고 있다.
Integer.compare()를 활용해 만약 o1과 o2의 절대 값이 같다면 실제 숫자를 기준으로 오름차순으로 정렬해주고, 아니라면 절대값을 기준으로 오름차순으로 정렬되도록 하였다.

 

💻 코드

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 IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());

        PriorityQueue<Integer> que = new PriorityQueue<>((o1, o2) ->
                Math.abs(o1) == Math.abs(o2) ? Integer.compare(o1, o2) : Integer.compare(Math.abs(o1), Math.abs(o2))
        );
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < N; i++) {
            int num = Integer.parseInt(br.readLine());

            if (num == 0)
                sb.append(que.size() == 0 ? 0 : que.poll()).append('\n');
            else que.add(num);
        }
        System.out.println(sb.toString());
    }
}

 

📊 제출 결과