BOJ 11286 - 절댓값 힙

www.acmicpc.net/problem/11286

11286번: 절댓값 힙

 

우선순위 큐에 less, greater 대신에 별도의

비교 연산을 위한 구조체를 정의해서 넣으면 된다.

 

- c++

 

#include <cmath>
#include <iostream>
#include <queue>
 
using namespace std;
 
int n;
 
struct compare {
    bool operator()(const int& num1, const int& num2) {
        if (abs(num1) != abs(num2))
            return abs(num1) > abs(num2);
        else
            return num1 > num2;
    }
};
 
int main() {
    cin.tie(0);
    ios_base::sync_with_stdio(0);
 
    priority_queue<int, vector<int>, compare> pq;
 
    cin >> n;
    for (int i = 0; i < n; ++i) {
        int x;
        cin >> x;
        if (x) {
            pq.push(x);
        } else {
            if (!pq.empty()) {
                cout << pq.top() << "\n";
                pq.pop();
            } else {
                cout << "0\n";
            }
        }
    }
 
    return 0;
}