백준 11286번 절댓값 힙

문제 링크입니다: https://www.acmicpc.net/problem/11286

앞서 푼 최소 힙, 최대 힙 문제보다는 까다로웠지만 pair<int, int> 형 힙을 사용하면 쉽게 풀 수 있는 문제였습니다.

이 문제의 경우 절댓값 최소 힙을 구현하는게 편할 것 같아 기존과는 다르게 디폴트 우선순위 큐에서 - 를 곱한 값을 넣도록 했습니다.

#include <iostream>

#include <vector>

#include <queue>

#include <algorithm>

#include <functional>

#include <map>

using namespace std;

 

map<int, int> visited;

 

int main(void)

{

        ios_base::sync_with_stdio(0);

        cin.tie(0);

        int N;

        cin >> N;

 

        priority_queue<pair<int, int>> pq;

        for (int i = 0; i < N; i++)

        {

                 int num;

                 cin >> num;

 

                 if (!num && pq.empty())

                         cout << 0 << "\n";

                 else if (!num)

                 {

                         cout << -pq.top().second << "\n";

                         pq.pop();

                 }

                 else

                         pq.push({ -abs(num), -num });

        }

        return 0;

}

개발환경:Visual Studio 2017

지적, 조언, 질문 환영입니다! 댓글 남겨주세요~