11286번 절댓값 힙

문제 출처 : https://www.acmicpc.net/problem/11286

 

11286번: 절댓값 힙

 

 

 

풀이 과정

 

STL의 priority_queue에 functor를 구현해서 문제를 해결했습니다. 

 

절댓값의 크기가 다른 경우에는 절댓값을 기준으로 오름차순으로 정렬하고, 절댓값의 크기가 같은 경우에는 절댓값을 씌우지 않은 값을 기준으로 오름차순으로 정렬합니다.

 

 

 

 

소스 코드

#include <iostream>
#include<queue>
#include<cmath>
using namespace std;
 
int n;
 
 
//functor 구현
struct cmp { 
    bool operator()(int a, int b) {
        if(abs(a) == abs(b)) {
            return a > b; 
        }
        return abs(a) > abs(b);
    }
};
 
int main() {
    cin >> n;
    
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    priority_queue<int, vector<int>, cmp> pq;
 
    for(int i=0; i<n;i++) {
        int x;
        cin >> x;
        if( x==0) {
            if(pq.empty()) cout << "0\n";
            else
            cout << pq.top() << "\n", pq.pop();
         
 
        } else {
            pq.push(x);
         
        } 
    }
 
}