linux-kernel-module-cheat/userland/cpp/bst_vs_heap.cpp

// https://github.com/cirosantilli/linux-kernel-module-cheat#bst-vs-heap

#include <algorithm>
#include <iostream>
#include <queue>
#include <random>
#include <set>

#include <lkmc/m5ops.h>

int main(int argc, char **argv) {
    typedef uint64_t I;
    std::vector<I> randoms;
    size_t i, n;
    std::priority_queue<I> heap;
    std::set<I> bst;
    unsigned int seed = std::random_device()();

    // CLI arguments.
    if (argc > 1) {
        n = std::stoi(argv[1]);
    } else {
        n = 1;
    }

    // Action.
    for (i = 0; i < n; ++i) {
        randoms.push_back(i);
    }
    std::shuffle(randoms.begin(), randoms.end(), std::mt19937(seed));
    for (i = 0; i < n; ++i) {
        auto random = randoms[i];

        // Heap.
        LKMC_M5OPS_RESETSTATS;
        heap.emplace(random);
        LKMC_M5OPS_DUMPSTATS;

        // BST.
        LKMC_M5OPS_RESETSTATS;
        bst.insert(random);
        LKMC_M5OPS_DUMPSTATS;
    }
}