当前位置:   article > 正文

数据结构 —— 堆_堆数据结构

堆数据结构

1.堆的概念及结构

堆是一种特殊的树形数据结构,称为“二叉堆”(binary heap)

看它的名字也可以看出堆与二叉树有关系:其实堆就是一种特殊的二叉树

堆的性质:

  • 堆中某个结点的值总是不大于或不小于其父结点的值;
  • 堆总是一棵完全二叉树

1.1大堆

大堆:

  • 大堆的根节点是整个堆中最大的数
  • 每个父节点的值都大于或等于其孩子节点的值
  • 每个父节点的孩子之间并无直接的大小关系

1.2小堆 

小堆:

  •  小堆的根节点是整个堆中最小的数
  • 每个父节点的值都小于或等于其孩子节点的值
  • 每个父节点的孩子之间并无直接的大小关系

 2.堆的实现

2.1使用数组结构实现的堆

由于堆是一个完全二叉树,所以堆通常使用数组来进行存储

使用数组的优点:

  • 相较于双链表更加的节省内存空间
  • 相较于单链表可以更好的算父子关系,并找到想要找的父子

2.2堆向上调整算法

堆的向上调整(也称为堆化、堆的修复或堆的重新堆化)是堆数据结构维护其性质的关键操作之一

现在我们给出一个数组,逻辑上看做一颗完全二叉树。我们通过从最后一个叶子节点开始的向上调整算法可以把它调整成一个小堆
向下调整算法有一个前提:最后一个叶子之前是一个堆才能调整

 int arr = [ 15, 18, 19, 25, 28, 34, 65, 49, 37, 10]

小堆演示向上调整算法演示过程

向上调整的过程 :将新插入的值与它的父亲相比,如果小则向上调整,调整完成后与新的父亲去比较,直到其值 >= 父亲的时候停止调整 

  1. void Swaps(HPDataType* a, HPDataType* b) {
  2. HPDataType temp;
  3. temp = *a;
  4. *a = *b;
  5. *b = temp;
  6. }
  7. //向上调整(小堆)
  8. //child是下标
  9. void AdjustUp(HPDataType* a, int child) {
  10. assert(a);
  11. int parent = (child - 1) / 2;//算父亲节点的下标
  12. //向下调整主要逻辑
  13. while (child > 0) //当调整至根节点时,已经调整至极限,不用在调整
  14. {
  15. //当父亲节点 > 孩子时,开始调整
  16. if (a[parent] > a[child])
  17. {
  18. Swaps(&a[child],&a[parent]); //交换
  19. child = parent; //走到新的位置为新一轮的向下调整做准备
  20. parent = (child - 1) / 2; //算出新位置的父亲节点下标
  21. }
  22. //当父亲节点 < 孩子时,说明调整已经完毕,退出循环
  23. else
  24. {
  25. break;
  26. }
  27. }
  28. }

2.3堆向下调整算法

在堆排序或其他需要维护堆性质的场景中,当堆的某个节点不满足堆的性质(对于最大堆,父节点小于其子节点;对于最小堆,父节点大于其子节点)时,就需要通过向下调整来修复这个子树,使其重新成为堆

现在我们给出一个数组,逻辑上看做一颗完全二叉树。我们通过从根结点开始的向下调整算法可以把它调整成一个小堆
向下调整算法有一个前提:左右子树必须是一个堆,才能调整
int array[] = {27,15,19,18,28,34,65,49,25,37};

 2.4堆的插入

堆的插入(HeapPush):通常通过将新元素添加到堆的末尾,并通过向上调整算法来维持堆的性质 (由于插入前的堆肯定是一个标准的堆,所以我们在将数据插入后执行一次向上调整算法,即可完成堆的插入)

2.5堆的删除

删除元素(HeapPop):在最大堆或最小堆中,通常删除的是根节点(即最大或最小元素),并通过向下调整算法来维持堆的性质 (由于删除前的堆肯定是一个标准的堆即左右子树肯定也是标准的堆,所以我们在将数据删除后执行一次向下调整算法,即可完成堆的删除)

为什么要删除根节点?

  • 相较于删除别的位置的节点,每次删除的根节点都是堆中最大或最小的数(大堆为最大,小堆为最小)、
  • 从根节点开始删除并调整堆结构,在实现上相对简便。只需删除后算法向下调整即可

2.6堆的代码实现

Heap.h

  1. #pragma once
  2. #include<stdio.h>
  3. #include<stdlib.h>
  4. #include<assert.h>
  5. typedef int HPDataType;
  6. typedef struct Heap
  7. {
  8. HPDataType* _a;
  9. int _size;
  10. int _capacity;
  11. }Heap;
  12. //堆的初始化
  13. void HeapInit(Heap* php);
  14. // 堆的销毁
  15. void HeapDestory(Heap* hp);
  16. // 堆的插入
  17. void HeapPush(Heap* hp, HPDataType x);
  18. // 堆的删除
  19. void HeapPop(Heap* hp);
  20. // 取堆顶的数据
  21. HPDataType HeapTop(Heap* hp);
  22. // 堆的数据个数
  23. int HeapSize(Heap* hp);
  24. // 堆的判空
  25. int HeapEmpty(Heap* hp);
  26. //向上调整
  27. void AdjustUp(HPDataType* a, int child);
  28. //向下调整
  29. void AdjustDown(HPDataType* a, int n, int parent);

Heap.c 

  1. //堆的初始化
  2. void HeapInit(Heap* hp) {
  3. assert(hp);
  4. hp->_a = NULL;
  5. hp->_capacity = hp->_size = 0;
  6. }
  7. // 堆的销毁
  8. void HeapDestory(Heap* hp) {
  9. assert(hp);
  10. free(hp->_a);
  11. hp->_capacity = hp->_size = 0;
  12. }
  13. // 堆的插入
  14. void HeapPush(Heap* hp, HPDataType x) {
  15. assert(hp);
  16. //扩容
  17. if (hp->_size == hp->_capacity)
  18. {
  19. int newcapacity = hp->_capacity == 0 ? 2 : hp->_capacity * 2;
  20. HPDataType* newa = (HPDataType*)realloc(hp->_a, newcapacity * sizeof(HPDataType));
  21. if (newa == NULL)
  22. {
  23. perror("realloc");
  24. return;
  25. }
  26. hp->_capacity = newcapacity;
  27. hp->_a = newa;
  28. }
  29. //插入数据
  30. hp->_a[hp->_size] = x;
  31. hp->_size++;
  32. //向上调整
  33. AdjustUp(hp->_a,hp->_size-1);
  34. }
  35. void Swaps(HPDataType* a, HPDataType* b) {
  36. HPDataType temp;
  37. temp = *a;
  38. *a = *b;
  39. *b = temp;
  40. }
  41. //向上调整(小堆)
  42. //child是数组的下标
  43. void AdjustUp(HPDataType* a, int child) {
  44. assert(a);
  45. int parent = (child - 1) / 2;
  46. while (child > 0)
  47. {
  48. if (a[parent] > a[child])
  49. {
  50. Swaps(&a[child],&a[parent]);
  51. child = parent;
  52. parent = (child - 1) / 2;
  53. }
  54. else
  55. {
  56. break;
  57. }
  58. }
  59. }
  60. // 堆的删除
  61. void HeapPop(Heap* hp) {
  62. assert(hp);
  63. assert(hp->_size);
  64. //删除顶部数据 ,先与末尾的交换,在向下调整
  65. Swaps(&hp->_a[0],&hp->_a[hp->_size-1]);//让数组首元素,与尾元素交换位置
  66. hp->_size--;
  67. AdjustDown(hp->_a, hp->_size, 0);
  68. }
  69. //向下调整(小堆)
  70. //n是数据数个数
  71. void AdjustDown(HPDataType* a, int n, int parent) {
  72. assert(a);
  73. //假设法,默认两个孩子最小的是左孩子
  74. int child = parent * 2 + 1;
  75. //当没有左孩子的时候停止向下调整,拿新算的孩子位置去判断
  76. while (child < n)
  77. {
  78. if (child + 1 < n && a[child + 1] < a[child])//挑最小的孩子换,且要注意有没有右孩子
  79. {
  80. child += 1;
  81. }
  82. if (a[child] < a[parent])//孩子比父亲小就往上换
  83. {
  84. Swaps(&a[child], &a[parent]);
  85. parent = child;//孩子变成父亲,与他的孩子比
  86. child = parent * 2 + 1;
  87. }
  88. else
  89. {
  90. break;
  91. }
  92. }
  93. }
  94. // 取堆顶的数据
  95. HPDataType HeapTop(Heap* hp) {
  96. assert(hp);
  97. assert(hp->_size);
  98. return hp->_a[0];
  99. }
  100. // 堆的数据个数
  101. int HeapSize(Heap* hp) {
  102. assert(hp);
  103. return hp->_size;
  104. }
  105. // 堆的判空
  106. int HeapEmpty(Heap* hp) {
  107. return hp->_size == 0;
  108. }

3堆的应用 — 堆排序 

堆排序,我们肯定是运用堆这个数据结构来完成我们的堆排序

接下来我们将充分的了解堆排序的运作原理

下面展示的是,用一个大堆如何排一个升序?

不难看出

  • 在每次交换时,堆顶最小的数都会沉到当前堆底
  • 小堆在经历过N(数据个数)轮后就会得到一个升序的数组
  • 大堆在经历过N(数据个数)轮后就会得到一个降序的数组

知道了堆排序的运转过程之后还有一个问题:使用者不可能说给你一个堆结构让你排序,肯定给的是一串无序且不是堆的数组给你排,这时侯我们就要考虑如何建堆了

3.1建堆

难道说建堆要用到上面写的堆结构,一个一个的去push吗?

其实不然,我们只需要使用向上调整算法向下调整算法就可以完成建堆

向上调整建堆法

1.构建过程

  • 初始时,将数组的第一个元素视为堆的根节点(对于下标从0开始的数组,根节点的下标为0)
  • 对于数组中剩余的元素(从下标1开始),将它们逐个视为“新插入”的元素,并执行向上调整操作
  • 在向上调整过程中,对于当前元素,首先计算其父节点的下标(parent = (child - 1) / 2)。然后,比较当前元素与其父节点的值
  • 如果当前元素的值大于其父节点的值(对于大根堆),则交换它们的位置。然后,将当前元素设置为新交换位置的父节点,并重复上述步骤,直到当前元素的值不大于其父节点的值或已经到达根节点
  • 通过重复上述步骤,直到所有元素都被处理过,最终得到的数组将满足堆的性质

2.时间复杂度

  • 向上调整建堆法的时间复杂度为O(N * logN),其中N是数组中的元素数量
  1. void Swaps(int* a, int* b) {
  2. int temp;
  3. temp = *a;
  4. *a = *b;
  5. *b = temp;
  6. }
  7. //向上调整(小堆)
  8. void AdjustUp(int* a, int child) {
  9. assert(a);
  10. int parent = (child - 1) / 2;
  11. while (child > 0)
  12. {
  13. if (a[parent] > a[child])
  14. {
  15. Swaps(&a[child], &a[parent]);
  16. child = parent;
  17. parent = (child - 1) / 2;
  18. }
  19. else
  20. {
  21. break;
  22. }
  23. }
  24. }
  25. //堆排序
  26. void HeapSort(int* a, int n) {
  27. //创建堆,向上调整建堆
  28. for (int i = 1; i < n; i++)
  29. {
  30. AdjustUp(a,i);
  31. }
  32. }

向下调整建堆法

向下调整(Adjust Down)是指从给定的非叶子节点开始,通过与其子节点比较并交换位(如果需要)来确保堆的性质

1.构建过程

  1. 确定开始位置
    • 对于长度为n的数组,由于堆是完全二叉树,所以最后一个非叶子节点的下标为(n-1-1)/2(整数除法)
    • 从这个下标开始,向前遍历所有非叶子节点
  2. 执行向下调整
  3. 遍历结束
    • 当所有非叶子节点都经过向下调整后,整个数组就形成了一个堆

2.时间复杂度

向下调整建堆法的时间复杂度为O(N),其中N是数组中的元素数量

  1. void Swaps(int* a, int* b) {
  2. int temp;
  3. temp = *a;
  4. *a = *b;
  5. *b = temp;
  6. }
  7. //向上调整(小堆)
  8. void AdjustUp(int* a, int child) {
  9. assert(a);
  10. int parent = (child - 1) / 2;
  11. while (child > 0)
  12. {
  13. if (a[parent] > a[child])
  14. {
  15. Swaps(&a[child], &a[parent]);
  16. child = parent;
  17. parent = (child - 1) / 2;
  18. }
  19. else
  20. {
  21. break;
  22. }
  23. }
  24. }
  25. //堆排序
  26. void HeapSort(int* a, int n) {
  27. //创建堆,向下调整建堆
  28. int parent = (n - 1 - 1) / 2; //找到最后一个非叶子节点
  29. for (parent; parent >= 0; parent--)
  30. {
  31. AdjustDown(a, n, parent);
  32. }
  33. }

3.2 利用堆删除思想来进行排序

  1. void Swaps(int* a, int* b) {
  2. int temp;
  3. temp = *a;
  4. *a = *b;
  5. *b = temp;
  6. }
  7. //向上调整(小堆)
  8. void AdjustUp(int* a, int child) {
  9. assert(a);
  10. int parent = (child - 1) / 2;
  11. while (child > 0)
  12. {
  13. if (a[parent] > a[child])
  14. {
  15. Swaps(&a[child], &a[parent]);
  16. child = parent;
  17. parent = (child - 1) / 2;
  18. }
  19. else
  20. {
  21. break;
  22. }
  23. }
  24. }
  25. //向下调整(小堆)
  26. void AdjustDown(int* a, int n, int parent) {
  27. assert(a);
  28. int child = parent * 2 + 1;
  29. while (child < n)
  30. {
  31. if (child + 1 < n && a[child + 1] < a[child])
  32. {
  33. child += 1;
  34. }
  35. if (a[child] < a[parent])
  36. {
  37. Swaps(&a[child], &a[parent]);
  38. parent = child;
  39. child = parent * 2 + 1;
  40. }
  41. else
  42. {
  43. break;
  44. }
  45. }
  46. }
  47. //堆排序
  48. void HeapSort(int* a, int n) {
  49. 创建堆,向上调整建堆
  50. //for (int i = 1; i < n; i++)
  51. //{
  52. // AdjustUp(a, i);
  53. //}
  54. //创建堆,向下调整建堆
  55. int parent = (n - 1 - 1) / 2;
  56. for (parent; parent >= 0; parent--)
  57. {
  58. AdjustDown(a, n, parent);
  59. }
  60. //小堆,可以排降序
  61. while (n)
  62. {
  63. Swaps(&a[0], &a[n - 1]);
  64. //交换完成把除了最后一个数据之外的数组看成一个新的堆,开始向下交换,形成新的小堆
  65. n--;
  66. AdjustDown(a, n, 0);
  67. }
  68. }

3.3建堆的时间复杂度 

树的高度

因为堆是完全二叉树,而满二叉树也是完全二叉树,此处为了简化使用满二叉树来证明 ( 时间复杂度本来看的 就是近似值,多几个结点不影响最终结果)

 

向下建堆时间复杂度

向下建堆的时间复杂度 = O(N)

向上建堆时间复杂度

向下建堆的时间复杂度 = O(N * logN)

4堆的应用 — Top-K问题

TOP-K 问题:即求数据结合中前 K 个最大的元素或者最小的元素,一般情况下数据量都比较大
比如:专业前 10 名、世界 500 强、富豪榜、游戏中前 100 的活跃玩家等
对于 Top-K 问题,能想到的最简单直接的方式就是排序,但是:如果数据量非常大,排序就不太可取了 ( 可能 数据都不能一下子全部加载到内存中) 。最佳的方式就是用堆来解决,基本思路如下:
1. 用数据集合中前 K 个元素来建堆
  • k个最大的元素,则建小堆
  • k个最小的元素,则建大堆
2. 用剩余的 N-K 个元素依次与堆顶元素来比较,不满足则替换堆顶元素
将剩余 N-K 个元素依次与堆顶元素比完之后,堆中剩余的 K 个元素就是所求的前 K 个最小或者最大的元素。

  1. void Swaps(int* a, int* b) {
  2. int temp;
  3. temp = *a;
  4. *a = *b;
  5. *b = temp;
  6. }
  7. //向下调整(小堆)大的下去
  8. //n是数据数个数
  9. void AdjustDown(HPDataType* a, int n, int parent) {
  10. assert(a);
  11. int child = parent * 2 + 1;
  12. while (child < n)
  13. {
  14. if (child + 1 < n && a[child + 1] < a[child])
  15. {
  16. child += 1;
  17. }
  18. if (a[child] < a[parent])
  19. {
  20. Swaps(&a[child], &a[parent]);
  21. parent = child;
  22. child = parent * 2 + 1;
  23. }
  24. else
  25. {
  26. break;
  27. }
  28. }
  29. }
  30. void CreateNDate()
  31. {
  32. // 造数据
  33. int n = 10000;
  34. srand((unsigned int)time(NULL));
  35. const char* file = "data.txt";
  36. FILE* fin = fopen(file, "w");
  37. if (fin == NULL)
  38. {
  39. perror("fopen error");
  40. return;
  41. }
  42. for (size_t i = 0; i < n; ++i)
  43. {
  44. int x = rand() % 1000000;
  45. fprintf(fin, "%d\n", x);
  46. }
  47. fclose(fin);
  48. }
  49. void PrintTopK(int k) {
  50. //找出前K个最大的数
  51. //打开文件
  52. FILE* p = fopen("data.txt", "r");
  53. if (p == NULL)
  54. {
  55. perror("fopen error");
  56. return;
  57. }
  58. //构建一个小堆
  59. int x = 0;
  60. int arr[10] = { 0 };
  61. for (int i = k; i < 10; i++)
  62. {
  63. fscanf(p,"%d", &x);
  64. arr[i] = x;
  65. }
  66. //创建堆,向下调整建堆,F(N)
  67. int parent = (k - 1 - 1) / 2;
  68. for (parent; parent >= 0; parent--)
  69. {
  70. AdjustDown(arr, k, parent);//这里的n数组的位置,里面的child会算出超过数组的位置,这样会停下来
  71. }
  72. //在将后面的数字依次对比小堆顶部,比它大就向下调整
  73. while (fscanf(p, "%d", &x) > 0)
  74. {
  75. if (arr[0] < x)
  76. {
  77. arr[0] = x;
  78. AdjustDown(arr, k, 0);
  79. }
  80. }
  81. for (int i = 0; i < k; i++)
  82. {
  83. printf("%d\n", arr[i]);
  84. }
  85. }
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/木道寻08/article/detail/878060
推荐阅读
相关标签
  

闽ICP备14008679号