[백준 / BOJ] C++ 11286 절대값 힙

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

11286번: 절댓값 힙

 

 

 

풀이

우선순위 큐의 greater의 자료형을 살펴보면 struct인 것을 알 수 있습니다.

절댓값을 구하는 operator를 struct으로 만들어 쉽게 해결할 수 있습니다.

 

Code

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

struct AbsGreater
{
	bool operator()(int a, int b) {
		if (abs(a) > abs(b))
			return true;
		else if (abs(a) == abs(b)) {
			if (a > b)
				return true;
			else
				return false;
		}
		else
			return false;
	}
};

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int N;

	cin >> N;

	priority_queue<int, vector<int>, AbsGreater> q;

	while (N--)
	{
		int x;

		cin >> x;

		if (x != 0)
		{
			q.push(x);
		}

		else if (x == 0)
		{
			if (q.empty() == true)
				cout << "0" << '\n';

			else {
				cout << q.top() << '\n';
				q.pop();
			}
		}
	}
}