(C++) - 백준(BOJ) 11279번 : 최대 힙 답

www.acmicpc.net/problem/11279

11279번: 최대 힙

heap을 이용한 자료구조는 priority_queue(우선순위큐)입니다. 이는 최상단에 queue헤더를 추가함으로써 stl로 사용할 수 있습니다. 이를 사용해보는 간단한 예제 문제였습니다.

 

 

Code

1. STL priority_queue 사용

#include <bits/stdc++.h>
#define fastio ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
using namespace std;
int n;
priority_queue <int, vector<int>> maxHeap;

int main(){
    fastio;
    cin >> n;
    while(n--){
        int x;
        cin >> x;
        if(x) maxHeap.push(x);
        else {
            if(!maxHeap.size()) cout << 0;
            else cout << maxHeap.top(),maxHeap.pop();
            cout << '\n';
        }
    }
}

 

2. 배열 사용

#include <iostream>
#include <algorithm>
using namespace std;
int t,x,hn, heap[100001];
void push(int x)//자식노드에서 부모노드로 올라가며 확인
{
    heap[++hn] = x;
    for (int i = hn; i > 1; i /= 2)//아래에 있는것이 부모노드 보다 크다면 swap
    {
        if (heap[i / 2] < heap[i])
            swap(heap[i / 2], heap[i]);
        else
            break;
    }
}
int pop()//부모노드에서 자식노드로 내려가며 확인
{
    int ans = heap[1];//heap의 가장 처음이 제일 큰 수
    heap[1] = heap[hn];
    heap[hn--] = 0;
    for (int i = 1; i*2<=hn;)
    {
        if (heap[i] > heap[i * 2] && heap[i] > heap[i * 2 + 1])
            break;
        else if (heap[i * 2 + 1] < heap[i * 2])
        {
            swap(heap[i], heap[i * 2]);
            i = i * 2;
        }
        else
        {
            swap(heap[i], heap[i * 2 + 1]);
            i = i * 2 + 1;
        }
    }
    return ans;
}
int main() {
    cin >> t;
    while (t--)
    {
        int x;
        cin >> x;
        if (x == 0)
        {
            if (hn == 0)
                cout << '0' << '\n';
            else
                cout  << pop() << '\n';
        }
        else
            push(x);
    }
}