堆排序原理解析
堆排序是一个十分高效的排序方法,其时间复杂度为:建堆O(n),调整堆O(nlogn),且为原地排序,空间复杂度为O(n)。
堆排序的流程
首先,二叉堆是一棵完全二叉树,我们在这里以数组来存储堆,一个有n个元素的堆下标从0开始到n-1结束。在这种情况下,一个节点i的左孩子下标为2*i+1
,右孩子的下标为2*i+2
。这棵完全二叉树的第一个非叶子节点的下标为(n-1)/2
。堆排序的流程如下(以大根堆为例):
- 建堆:从第一个非叶子节点开始,从右至左,从上至下,一直调整堆,直到所有父节点的值都大于其子节点的值。(因为是从第一个非叶子节点开始的,所以能够保证当待调整节点调整到满足堆排序的节点时,剩下的节点肯定是符合堆排序的)。
- 循环将堆顶和待排序数组的最后一个元素交换,然后再从堆顶元素开始调整。这样末尾的元素就是最大值,也就实现了从小到大排序。
对应的代码
调整堆
从下标为index的节点出发,一直调整堆,直到父节点的值大于两个子节点的值。
void adjust(vector<int> & nums, int index,int n){
//n是待排序数组的末尾下标
//index为开始调整的节点的下标
//开始调整
int i=index;
int j=2*i+1