STC库中的优先级队列(pqueue)实现详解

汤璞亚Heath

于 2025-06-28 09:26:47 发布

阅读量307

点赞数 3

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/gitblog_00903/article/details/148971646

STC库中的优先级队列(pqueue)实现详解

STC A modern, user friendly, generic, type-safe and fast C99 container library: String, Vector, Sorted and Unordered Map and Set, Deque, Forward List, Smart Pointers, Bitset and Random numbers. 项目地址: https://gitcode.com/gh_mirrors/stc/STC

概述

优先级队列(Priority Queue)是一种特殊的容器适配器，它能够提供对最大(默认)或最小元素的快速访问，同时保证插入和删除操作的高效性。STC库中的pqueue模块实现了这一数据结构，其设计灵感来源于C++标准库中的std::priority_queue。

核心特性

时间复杂度：
- 查看顶部元素(top)：O(1)
- 插入元素(push)：O(log n)
- 删除顶部元素(pop)：O(log n)
自定义排序：
- 默认情况下，优先级队列维护最大堆特性
- 可通过i_cmp或i_less宏定义自定义比较逻辑
- 使用-c_default_cmp可实现最小堆

头文件与声明

使用STC的优先级队列需要包含头文件<stc/pqueue.h>，并通过宏定义配置队列行为：

// 基本定义方式
#define i_key <t>        // 指定元素类型
#define i_less <fn>      // 定义比较函数(必需，除非是基本类型)
#define i_cmp <fn>       // 三路比较函数的替代方案

// 高级配置选项
#define i_keydrop <fn>   // 自定义元素销毁函数
#define i_keyclone <fn>  // 自定义元素克隆函数
#define i_keyraw <t>     // 定义转换类型
#define i_keyfrom <fn>   // 从原始类型转换
#define i_keytoraw <fn>  // 转换为原始类型

主要API方法

初始化与销毁

pqueue_X_init()        // 创建空队列
pqueue_X_with_capacity() // 预分配容量的队列
pqueue_X_clone()       // 克隆队列
pqueue_X_drop()        // 销毁队列

容量操作

pqueue_X_size()        // 获取元素数量
pqueue_X_is_empty()    // 检查是否为空
pqueue_X_reserve()     // 预留空间
pqueue_X_shrink_to_fit() // 缩减容量

元素操作

pqueue_X_top()         // 访问顶部元素
pqueue_X_push()        // 插入元素
pqueue_X_emplace()     // 就地构造元素
pqueue_X_pop()         // 删除顶部元素
pqueue_X_pull()        // 取出并返回顶部元素
pqueue_X_make_heap()   // 堆化操作

实际应用示例

下面是一个使用优先级队列处理大量随机数的示例，展示了如何实现最小堆：

#include <stdio.h>
#include <stc/random.h>

// 定义最小堆优先级队列
#define T PriorityQ, int32
#define i_cmp -c_default_cmp // 负号表示最小堆
#include <stc/pqueue.h>

int main(void) {
    const isize N = 10000000;
    PriorityQ numbers = {0};

    // 插入五百万个随机数
    for (c_range(N/2))
        PriorityQ_push(&numbers, (int32)(crand32_uint() >> 6));

    // 插入一些特定负数
    int32 nums[] = {-231, -32, -873, -4, -343};
    for (c_range(i, c_countof(nums)))
        PriorityQ_push(&numbers, nums[i]);

    // 再插入五百万个随机数
    for (c_range(N/2))
        PriorityQ_push(&numbers, (int32)(crand32_uint() >> 6));

    // 输出最小的50个数
    for (c_range(50)) {
        printf("%d ", *PriorityQ_top(&numbers));
        PriorityQ_pop(&numbers);
    }
    
    PriorityQ_drop(&numbers);
}