Koala - 15기/코딩테스트 준비 스터디

[백준/C++] 11279번: 최대 힙

.우디. 2024. 8. 18. 20:09

문제&링크

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

 

풀이

1. 우선순위 큐를 이용하여 최대 힙을 만든다.

2. 0을 입력받은 순간 최대 힙이 비어있다면 0을 출력한다.

3. 0을 입력받은 순간 최대 힙이 비어있지 않다면 제일 큰 값, 즉 우선 순위 큐의 top에 해당하는 값을 출력한다.

4. 0이 아닌 다른 수를 입력받을 경우 최대 힙에 저장한다.

 

코드

#include <iostream>
#include <queue>

using namespace std;

priority_queue<int> pq;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    
    int N;
    cin >> N;
    
    for (int i = 0; i < N; i++) {
        int num;
        cin >> num;
        
        if (num == 0) {
            if (pq.empty()) cout << 0 << "\n";
            else {
                cout << pq.top() << "\n";
                pq.pop();
            }
        }
        else {
            pq.push(num);
        }
        
    }
}