백준 11286 (절대값 힙) / C++

링크 : 11286번: 절댓값 힙 (acmicpc.net)

11286번: 절댓값 힙

>풀이

자료구조 우선순위 큐에서 비교함수(구조체)를 어떻게 활용하는지를 묻는 문제입니다.

#include <bits/stdc++.h>
using namespace std;

struct cmp {
        bool operator()(int a, int b) {
                if (abs(a) == abs(b))
                        return a > b;
                else
                        return abs(a) > abs(b);
        }
};
priority_queue<int, vector<int>, cmp> pq;
int N, x;

int main(void)
{
        scanf("%d", &N);
        while (N--) {
                scanf("%d", &x);
                if (x == 0) {
                        if (pq.empty()) printf("0\n");
                        else {
                                printf("%d\n", pq.top());
                                pq.pop();
                        }
                }
                else {
                        pq.push(x);
                }
        }
        return 0;
}