[C/C++ 백준 11286번] 절댓값 힙 (Silver 1)

www.acmicpc.net/problem/11286

11286번: 절댓값 힙

직접 Comp 클래스를 만들어 연산자를 만들어 주어야 한다. 절댓값을 계산해서 같을 경우와 아닐 경우 리턴 값을 잘 생각해주자.

#include <cstdio>
#include <queue>
using namespace std;
int absf(int a){
	if(a>0)
		return a;
	return -a;
}
class Comp{
	public:
		bool operator()(const int &a, const int &b){
			if(absf(a)==absf(b))
				return a>b;
			return absf(a)>absf(b);
		}
};
int main(void){
	priority_queue <int, vector<int>, Comp> q;
	int N, x;
	scanf("%d", &N);
	for(int i=0; i<N; i++){
		scanf("%d", &x);
		if(x==0){
			if(q.empty())
				printf("0\n");
			else{
				printf("%d\n", q.top());
				q.pop();
			}
		}
		else
			q.push(x);
	}
}