堆排序是怎么排的?

我们先看看究竟什么是堆?以大顶堆为例:

对于一棵完全二叉树而言,当每个结点不小于其子结点时,便可称之为堆(大顶堆),比如:

 

 

原始的待排序的数组为:30, 20, 40, 10, 0, 60, 80, 70其对应的完全二叉树为:

 

 

接下来,我们来图解堆排序,并用程序来实现堆排序。在这个过程中,希望大家感受到堆之美。

 

图解堆排序

一. 构建堆

第1步:

如上图,最后一个非叶子结点是10,发现10比70小,所以70必须上浮,得到的结果为:

 

 

 

第2步:

如上图,倒数第二个非叶子结点为40,在40,60,80这三个数中,80最大,所以80必须上浮,得到的结果如下:

 

 

 

第3步:

如上图,倒数第三个非叶子结点为20,而20比70小,所以70必须上浮,20下沉后,发现比下面的10还大,所以没有必要沉底,得到的结果为:

 

 

 

第4步:

如上图,倒数第四个非叶子结点为30,在30,70,80中,80最大,所以80要上浮,30下沉。然而,30比60和40都小,所以要继续下沉,得到的结果是:

 

 

到此为止,可以看到,一个大顶堆已经形成,可以看到,最大的80已经被选择出来了。

 

二. 调整堆

我们把堆顶的最大值80调整到最后,保存下来,得到的结果是:

 

 

接下来的工作就是对上面红框中的的7个结点进行调整,使之形成新的堆。

很显然,根据之前调整的过程可知,两个蓝色框中的结点,已经分别成堆了,所以这次的调整就简单多了,直接瞄准待调整的10即可。

之前已经把8个结点调整成堆,那么调整上面红色框中的7个结点成堆便不在话下。于是,这7个结点中最大的70被调到了堆顶,如下:

 

 

80是最大的值,放在最后。堆顶的70是第二大的值,放在倒数第二的位置,所以跟40进行交换,得到的结果为:

 

 

可见,通过2次从堆顶摘下最大元素,分别把80和70选出来了。接下来,用相同的方法,把60选出来,依此循环,最后得到的二叉树为:

 

 

终于,实现了排序,这就是所谓的堆排序,其平均时间复杂度为O(N*logN), 比冒泡排序好多啦。

 

堆排序实现

接下来,我们用代码来实现堆排序,如下:

#include<iostream>
using namespace std;

void print(int a[], int n)
{
  int i;
  for(i = 0; i < n; i++)
  {
    cout << a[i] << " ";
  }

  cout << endl;
}
 
void heapAdjust(int a[], int low, int high)
{
  int pivotKey = a[low - 1];
  int i;
  for(i = 2 * low; i <= high; i *= 2) 
  {
    if(i < high && a[i - 1] < a[i])
    {
      i++; //i指向较大值
    }

    if(pivotKey >= a[i - 1])
    {
      break;
    }
  
    a[low - 1] = a[i - 1];
    low = i; 
  }
  a[low - 1] = pivotKey; 
}
 
void heapSort(int a[], int n)
{
  int i, tmp;
  for(i = n/2; i > 0; i--) 
  {
    heapAdjust(a, i, n);
    print(a, n);
  }

  for(i = n; i > 1; i--)
  {
    tmp = a[i -1];
    a[i - 1] = a[0];
    a[0] = tmp;
    heapAdjust(a, 1, i - 1); 
    print(a, n);
  }
}

int main()
{
  int a[] = {30, 20, 40, 10, 0, 60, 80, 70};
  int n = sizeof(a) / sizeof(a[0]);
  heapSort(a, n);
  print(a, n);
  return 0;
}

 

最终的排序结果如下:

0 10 20 30 40 60 70 80 

 

堆是一种重要的数据结构,堆排序也是非常重要的算法。在笔试面试中,经常考到堆的相关应用。

posted @ 2021-07-28 10:18  苍青浪  阅读(4416)  评论(0编辑  收藏  举报