数据结构——堆排序详解

x33g5p2x  于2021-09-25 转载在 其他  
字(1.6k)|赞(0)|评价(0)|浏览(426)

1.堆的概念及结构

如果有一个关键码的集合K = {k0,k1, k2,…,kn-1},把它的所有元素按完全二叉树(二叉树具体概念参见——二叉树详解)的顺序存储方式存储在一个一维数组中,并满足:Ki <= K2i+1 且 Ki<= K2i+2 (Ki >= K2i+1 且 Ki >= K2i+2) i = 0,1,2…,则称为小堆(或大堆)。将根节点最大的堆叫做最大堆或大根堆,根节点最小的堆叫做最小堆或小根堆。

堆的性质:

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

2.堆的实现

堆的实现请参见——二叉树详解(堆的实现)

2.1 堆的向下调整算法

(此文章都已建小堆为例)
向下调整算法前提:当前树左右子树都是小堆

核心思想:选出左右孩子中小的那个,和父亲交换,小的往上浮,大的往下沉,这里是小堆,如果是大堆则相反。

代码实现

  1. void swap(int *x, int *y)
  2. {
  3. int temp = *x;
  4. *x = *y;
  5. *y = temp;
  6. }
  7. //堆向下调整算法
  8. void AdjustDown(int *a, int n, int root)
  9. {
  10. int parent = root;
  11. int child = parent * 2 + 1;
  12. while (child<n)
  13. {
  14. //保证孩子节点child为两个孩子中的最小值;保证不越界
  15. if (a[child] > a[child + 1] && child+1 < n)
  16. ++child;
  17. if (a[child] < a[parent])
  18. {
  19. swap(&a[child], &a[parent]);
  20. parent = child;
  21. child = parent * 2 + 1;
  22. }
  23. else
  24. break;
  25. }
  26. }

2.2 堆的向上调整算法

使用场景:向上调整算法适用于向堆中插入数据,当向堆中插入数据就可能会导致堆失去大堆或者小堆的性质,此时需要重新调整,向上调整的思路与向下调整算法的思路类似,向上调整算法只需要从插入结点位置开始和父节点比较。
图示:

代码实现:

  1. void AdjustUp(int *a, int child)
  2. {
  3. int parent = (child - 1) / 2;
  4. while (child > 0)
  5. {
  6. if (a[parent] > a[child])
  7. {
  8. swap(&a[parent], &a[child]);
  9. child = parent;
  10. parent = (child - 1) / 2;
  11. }
  12. else
  13. break;
  14. }
  15. }

2.3 建堆(数组)

从最后一个非叶子节点位置行依次开始调整,如图:

代码实现:

  1. int parent = (n-2) / 2;
  2. //首先对每一个非叶子节点进行一次向下调整算法,保证每个非叶子节点的
  3. //孩子都小于它的父节点,然后可得到最小值,就在堆的顶端的父节点(也叫做建小堆)
  4. while (parent >= 0)
  5. {
  6. AdjustDown(a, n, parent);
  7. --parent;
  8. }

2.4 堆排序

升序建大堆,降序建小堆

  1. void HeapSort(int *a, int n)
  2. {
  3. int parent = (n-2) / 2;
  4. //首先对每一个非叶子节点进行一次向下调整算法,保证每个非叶子节点的
  5. //孩子都小于它的父节点,然后可得到最小值,就在堆的顶端的父节点(也叫做建小堆)
  6. while (parent >= 0)
  7. {
  8. AdjustDown(a, n, parent);
  9. --parent;
  10. }
  11. int end = n-1;
  12. while (end>0)
  13. {
  14. //将堆顶的数与最后的end,以此循环,进行交换就可得到有序序列
  15. //注意:建小堆,得到降序序列
  16. swap(&a[end], &a[0]);
  17. AdjustDown(a, end, 0);
  18. --end;
  19. }
  20. }

2.5 堆排序的时间复杂度

所以建堆时间复杂度为O(N);
向下调整算法时间复杂度 O(logN);
所以堆排序的时间复杂度为 O(N/*logN)

相关文章