[백준] 11286번 - 절댓값 힙 (Java)(◎)

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

11286번: 절댓값 힙

 

<풀이>

1. 입력값이 실시간으로 오른차순 정렬이 되어 있어야 한다.

2. 입력값이 양수와 정수가 번갈아 존재하고, 절대값 기준으로 정렬이 되어야 한다.

3. 절대값이 같을 경우 부호를 붙였을 때 작은 수가 앞쪽에 와야 한다.

4. PriorityQueue를 사용해서 실시간 정렬을 해보자.

5. class Node를 구성해서 num은 원래 값, abs는 절대값을 저장한다.

class Node implements Comparable<Node> {         int num;         int abs;         Node (int num, int abs) {             this.num = num;             this.abs = abs;         }         @Override         public int compareTo (Node o) {             if (this.abs == o.abs) // 절대값이 같을 경우 값이 작은 수가 앞으로 가게 정렬                 return this.num - o.num;             else  // 절대값이 다를 경우는 절대값이 작은 값이 앞으로 가게 정렬                 return this.abs - o.abs;         }      }

 

<구현>

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
import java.util.Arrays;
import java.util.PriorityQueue;
import java.lang.Comparable;

public class Main {
    static int N;
    static int A[];
    PriorityQueue<Node> PQ = new PriorityQueue<>();

    class Node implements Comparable<Node> {
        int num;
        int abs;
        Node (int num, int abs) {
            this.num = num;
            this.abs = abs;
        }
        @Override
        public int compareTo (Node o) {
            if (this.abs == o.abs) // 절대값이 같을 경우 값이 작은 수가 앞으로 가게 정렬
                return this.num - o.num;
            else  // 절대값이 다를 경우는 절대값이 작은 값이 앞으로 가게 정렬
                return this.abs - o.abs;
        } 
    }

	void Solve() {
        StringBuilder sb = new StringBuilder();
        for (int x : A) {
            if (x == 0) { // 출력 명령어 실행
                if (PQ.isEmpty()) {
                    sb.append(0+"\n");
                } else {
                    Node now = PQ.poll();
                    sb.append(now.num+"\n");
                }
            } else {
                PQ.offer(new Node(x, Math.abs(x)));
            }
        }
        System.out.println(sb.toString());
	}

    void inputData() throws Exception {
        InputStreamReader reader = new InputStreamReader(System.in);
        BufferedReader br = new BufferedReader(reader);
        N = Integer.parseInt(br.readLine());
        A = new int[N];
        //StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i=0; i<N; i++) {
            A[i] = Integer.parseInt(br.readLine());
        }
    }

    public static void main(String[] args) throws Exception {
        Main m = new Main();
        m.inputData(); // 입력 받는 부분
        m.Solve();// 여기서부터 작성
    }
}