hello-algo/ko/codes/cpp/chapter_heap/heap.cpp

/**
 * File: heap.cpp
 * Created Time: 2023-01-19
 * Author: LoneRanger([email protected])
 */

#include "../utils/common.hpp"

void testPush(priority_queue<int> &heap, int val) {
    heap.push(val); // 힙에 원소 삽입
    cout << "\n원소 " << val << " 힙에 삽입 후" << endl;
    printHeap(heap);
}

void testPop(priority_queue<int> &heap) {
    int val = heap.top();
    heap.pop();
    cout << "\n힙의 루트 원소 " << val << " 힙에서 꺼낸 후" << endl;
    printHeap(heap);
}

/* 실행 코드 */
int main() {
    /* 힙 초기화 */
    // 최소 힙 초기화
    // priority_queue<int, vector<int>, greater<int>> minHeap;
    // 최대 힙 초기화
    priority_queue<int, vector<int>, less<int>> maxHeap;

    cout << "\n다음 테스트 예제는 최대 힙이다" << endl;

    /* 힙에 원소 삽입 */
    testPush(maxHeap, 1);
    testPush(maxHeap, 3);
    testPush(maxHeap, 2);
    testPush(maxHeap, 5);
    testPush(maxHeap, 4);

    /* 힙의 루트 원소 반환 */
    int peek = maxHeap.top();
    cout << "\n힙의 루트 원소: " << peek << endl;

    /* 힙의 루트 원소 꺼내기 */
    testPop(maxHeap);
    testPop(maxHeap);
    testPop(maxHeap);
    testPop(maxHeap);
    testPop(maxHeap);

    /* 힙 크기 반환 */
    int size = maxHeap.size();
    cout << "\n힙의 원소 개수: " << size << endl;

    /* 힙이 비어 있는지 판단 */
    bool isEmpty = maxHeap.empty();
    cout << "\n힙이 비어 있는가: " << isEmpty << endl;

    /* 리스트를 입력하여 힙 구성 */
    // 시간 복잡도는 O(n)이며 O(nlogn)이 아니다
    vector<int> input{1, 3, 2, 5, 4};
    priority_queue<int, vector<int>, greater<int>> minHeap(input.begin(), input.end());
    cout << "리스트를 입력하여 최소 힙을 구성한 후" << endl;
    printHeap(minHeap);

    return 0;
}