深入探讨优先队列：原理、实现与应用

#include <iostream>
#include <queue>

int main() {
    // 默认情况下，std::priority_queue 是一个最大堆
    std::priority_queue<int> maxHeap;

    // 插入元素
    maxHeap.push(10);
    maxHeap.push(30);
    maxHeap.push(20);

    // 查看最大元素
    std::cout << "Top element: " << maxHeap.top() << std::endl; // 输出 30

    // 删除最大元素
    maxHeap.pop();
    std::cout << "Top element after pop: " << maxHeap.top() << std::endl; // 输出 20

    // 判断队列是否为空
    if (!maxHeap.empty()) {
        std::cout << "Priority queue is not empty." << std::endl;
    }

    // 获取队列大小
    std::cout << "Size of priority queue: " << maxHeap.size() << std::endl;

    return 0;
}

2、自定义比较函数

std::priority_queue默认是一个最大堆，即优先级最高的元素（最大值）位于堆顶。如果需要实现最小堆，可以通过自定义比较函数来实现。

cpp 复制代码

#include <iostream>
#include <queue>
#include <vector>
#include <functional> // 用于 std::greater

int main() {
    // 使用 std::greater<int> 作为比较函数，实现最小堆
    std::priority_queue<int, std::vector<int>, std::greater<int>> minHeap;

    // 插入元素
    minHeap.push(10);
    minHeap.push(30);
    minHeap.push(20);

    // 查看最小元素
    std::cout << "Top element: " << minHeap.top() << std::endl; // 输出 10

    // 删除最小元素
    minHeap.pop();
    std::cout << "Top element after pop: " << minHeap.top() << std::endl; // 输出 20

    return 0;
}

3、自定义数据结构

std::priority_queue也可以用于自定义数据结构。需要为自定义数据结构提供比较函数或重载比较运算符。

cpp 复制代码

#include <iostream>
#include <queue>
#include <vector>

struct Task {
    int priority;
    std::string name;

    // 重载小于运算符，用于最大堆
    bool operator<(const Task& other) const {
        return priority < other.priority;
    }

    // 重载大于运算符，用于最小堆
    bool operator>(const Task& other) const {
        return priority > other.priority;
    }
};

int main() {
    // 最大堆，根据 Task 的 priority 排序
    std::priority_queue<Task> maxHeap;

    maxHeap.push({10, "Task A"});
    maxHeap.push({30, "Task B"});
    maxHeap.push({20, "Task C"});

    std::cout << "Top task: " << maxHeap.top().name << " (Priority: " << maxHeap.top().priority << ")" << std::endl;

    // 最小堆，根据 Task 的 priority 排序
    std::priority_queue<Task, std::vector<Task>, std::greater<Task>> minHeap;

    minHeap.push({10, "Task A"});
    minHeap.push({30, "Task B"});
    minHeap.push({20, "Task C"});

    std::cout << "Top task: " << minHeap.top().name << " (Priority: " << minHeap.top().priority << ")" << std::endl;

    return 0;
}

五、优先队列的时间复杂度

插入操作（Push）：O(log n)
删除操作（Pop）：O(log n)
查看顶部元素（Top）：O(1)
判断队列是否为空（Empty）：O(1)
获取队列大小（Size）：O(1)

六、优先队列的应用场景

任务调度：根据任务的优先级动态调度任务。
图算法：如Dijkstra算法中用于选择最短路径的节点。
数据压缩：如Huffman编码中用于构建最优前缀码。
事件驱动模拟：如离散事件模拟中按时间顺序处理事件。

总结

优先队列是一种非常重要的数据结构，广泛应用于各种算法和系统中。C++中的std::priority_queue提供了简单易用的接口，并且可以通过自定义比较函数或重载运算符来适应不同的需求。理解优先队列的实现原理及其应用场景，对于编写高效的算法和系统至关重要。