堆与优先队列——练习题

嵌入式小小聪

已于 2024-04-27 16:38:08 修改

阅读量232

点赞数 3

分类专栏：数据结构与算法文章标签：数据结构

于 2024-04-27 16:13:09 首次发布

本文链接：https://blog.csdn.net/qq_68915288/article/details/138250991

版权

本文介绍了如何利用小顶堆数据结构实现在线求解数据流中的第K大元素，以及如何通过堆排序和哈希集合来计算数据流的中位数，并避免重复丑数的计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. 数据流中的第 K 大元素

代码实现：

思路：创建一个大小为 k 的小顶堆，堆顶元素就是第 K 大元素

typedef struct {
    int *__data, *data;
    int size;
    int n;
} KthLargest;

#define swap(a, b) { \
    __typeof(a) __c = (a); \
    (a) = (b); \
    (b) = __c; \
}

// 从1开始存储，根结点下标为1
#define FATHER(i) ((i) / 2)
#define LEFT(i) ((i) * 2)
#define RIGHT(i) ((i) * 2 + 1)

// 向上调整(非递归)
void up_update(int *data, int i) {
    while (i > 1 && data[i] < data[FATHER(i)]) {
        swap(data[i], data[FATHER(i)]);
        i = FATHER(i);
    }
}

// 向下调整
void down_update(int *data, int i, int n) {
    while (LEFT(i) <= n) {
        int ind = i, l = LEFT(i), r = RIGHT(i);
        if (data[l] < data[ind]) {
            ind = l;
        }
        if (r <= n && data[r] < data[ind]) {
            ind = r;
        }
        if (ind == i) {
            break;
        }
        swap(data[i], data[ind]);