백준 11279 - 최대 힙

문제

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

11279번: 최대 힙

 

 

풀이

최대 힙을 만드는 문제이다. stl의 우선순위 큐를 이용한다.

우선순위 큐를 이용하려면 #include<queue>를 해준다.

우선순위 큐는 priority_queue<자료형, 구현체, 비교 연산자>로 정의한다.

priority_queue는 기본적으로 큰 값의 우선순위가 높으므로 따로 비교 연산자를 지정하지 않아도 된다.

 

입력하는 값이 0이면 우선순위 큐에서 우선순위가 가장 높은 값을 출력하고, 0이 아니면 우선순위 큐에 저장을 한다.

 

 

코드

#include <iostream>
#include <queue>
 
using namespace std;
 
int N, num;
priority_queue<int> pq;
 
int main()
{
    cin.tie(0);
    ios::sync_with_stdio(false);
 
    cin >> N;
 
    for (int i = 0; i < N; i++)
    {
        cin >> num;
        if (num) pq.push(num);
        else
        {
            if (pq.empty()) cout << 0 << '\n';
            else
            {
                cout << pq.top() << '\n';
                pq.pop();
            }
        }
    }
 
    return 0;
}