[数据结构 -- 手撕排序算法第四篇] 堆排序,一篇带你搞懂堆排序

简介: [数据结构 -- 手撕排序算法第四篇] 堆排序,一篇带你搞懂堆排序

1、堆的应用 -- 堆排序

堆是一个完全二叉树,完全二叉树用数组存储数据最优。
堆排序即利用堆的思想来进行排序,总共分为两个步骤:

1、建堆

升序:建大堆

降序:建小堆

2、利用堆删除的思想来进行排序

建堆和堆删除中都用到了向下调整,因此掌握了向下调整,就可以完成堆排序。

1.1 堆排序的思路分析

我们本篇文章使用小堆进行讲解,小堆排序是降序。

对堆还不是很了解的同学可以浅看一下堆的那篇文章:戳这里即可跳转
1、我们先对数组里的元素进行向下调整建成小堆;


2、小堆的堆顶元素肯定是数组中最小的,因此我们将堆顶元素(数组首元素)与数组尾元素交换,将数组尾元素不在看作是数组中的元素(size--),再从堆顶开始向下调整重新构建小堆,不断重复就可以实现降序排序(升序只要将小堆改为大堆就可以实现)。

2、建堆

我们建堆可以使用向上调整建堆,也可以使用向下调整建堆,我们该如何选择呢?

那肯定是谁的时间复杂度小我就选谁,那接下来我们分析一下两种建堆的时间复杂度:

2.1 向上调整建堆:O(N*logN)

2.1.1 向上调整代码

void AdjustUp(HPDataType* a, int child)
{
  int parent = (child - 1) / 2;
  while (child > 0)
  {
    if (a[child] < a[parent])//这里控制大小堆
    {
      Swap(&a[child], &a[parent]);
      child = parent;
      parent = (child - 1) / 2;
    }
    else
    {
      break;
    }
  }
}

2.1.2 向上调整建堆代码

//建堆 -- 向上调整,时间复杂度:O(N*log(N))
for (int i = 0; i < size; i++)
{
  AdjustUp(a, i);
}

我们画图来分析一下向上调整建堆时间复杂度:

堆是一个完全二叉树,满二叉树也是完全二叉树,因此我们以满二叉树为例推出向上调整建堆的时间复杂度为O(N*logN)。

2.2 向下调整建堆:O(N)

2.2.1 向下调整代码

void AdjustDown(HPDataType* a, int size, int parent)
{
  int child = parent * 2 + 1;
  while (child < size)//当child大于了数组大小就跳出循环
  {
    //找出左右孩子中小/大的那个(假设法)
    if (child + 1 < size && a[child + 1] < a[child])
    {
      child++;
    }
    if (a[child] < a[parent])
    {
      Swap(&a[parent], &a[child]);
      parent = child;
      child = parent * 2 + 1;
    }
    else
    {
      break;
    }
  }
}

2.2.2 向下调整建堆代码

for (int i = (size - 1 - 1) / 2; i >= 0; i--)
{
    AdjustDown(a, size, i);
}

我们画图来分析一下向下调整建堆时间复杂度:

向下调整建堆时间复杂度:O(N)。

如此分析下来我们就可以知道,向下调正建堆才是最优选择。

3、堆排序实现代码

//堆排序时间复杂度O(N + N*logN)
void HeapSort(int* a, int size)
{
  //升序 -- 建大堆
  //降序 -- 建小堆
  //建堆 -- 向上调整,时间复杂度:O(N*log(N))
  //for (int i = 0; i < size; i++)
  //{
  //  AdjustUp(a, i);
  //}
  //建堆 -- 向下调整,时间复杂度:O(N)
  //倒着调整
  //叶子节点不需要处理
  //倒数第一个非叶子节点:最后一个节点的父亲开始调整
  for (int i = (size - 1 - 1) / 2; i >= 0; i--)
  {
    AdjustDown(a, size, i);
  }
  //O(N*log(N))
  int end = size - 1;
  while (end)
  {
    //1.先交换
    Swap(&a[0], &a[end]);
    //2.再调整,选出当前大小的数组中最小数
    AdjustDown(a, end, 0);
    end--;
  }
}

4、堆排序测试

我们建的是小堆,因此最终排的是降序。


*** 本篇结束 ***

相关文章
|
23天前
|
存储 算法 索引
【算法与数据结构】队列的实现详解
【算法与数据结构】队列的实现详解
|
26天前
|
算法
【算法与数据结构】二叉树(前中后)序遍历2
【算法与数据结构】二叉树(前中后)序遍历
|
3天前
|
人工智能 搜索推荐 算法
sort-04-heap sort 堆排序算法详解
这是一个关于排序算法的系列文章摘要,包括了10篇关于不同排序算法的链接,如冒泡排序、快速排序、选择排序、堆排序等。堆排序是一种基于比较的排序算法,利用了近似完全二叉树的结构并满足最大堆或最小堆的性质。最大堆中,每个节点的值都大于或等于其子节点。文章详细解释了最大堆的概念、节点访问方式以及堆的操作,包括堆调整和堆排序的过程,并通过图解展示了堆排序的具体步骤。此外,还提供了一个Java实现的堆排序代码示例。整个排序系列来源于一个开源项目,读者可以通过链接查看完整内容。
sort-04-heap sort 堆排序算法详解
|
5天前
|
存储 算法
数据结构开篇(普普通通浅浅聊数据结构)什么是数据结构 、什么是算法、重要性、如何学好数据结构呢
数据结构开篇(普普通通浅浅聊数据结构)什么是数据结构 、什么是算法、重要性、如何学好数据结构呢
|
13天前
|
算法
堆排序+TopK问题——“数据结构与算法”
堆排序+TopK问题——“数据结构与算法”
|
15天前
|
存储 机器学习/深度学习 算法
上机实验三 图的最小生成树算法设计 西安石油大学数据结构
上机实验三 图的最小生成树算法设计 西安石油大学数据结构
19 1
|
23天前
|
算法 索引
【算法与数据结构】深入二叉树实现超详解(全源码优化)
【算法与数据结构】深入二叉树实现超详解(全源码优化)
|
23天前
|
存储 算法
【算法与数据结构】深入解析二叉树(二)之堆结构实现
【算法与数据结构】深入解析二叉树(二)之堆结构实现
|
26天前
|
算法 C语言
【算法与数据结构】 C语言实现单链表队列详解2
【算法与数据结构】 C语言实现单链表队列详解
|
26天前
|
存储 算法 C语言
【算法与数据结构】 C语言实现单链表队列详解1
【算法与数据结构】 C语言实现单链表队列详解