堆排序笔记

堆排序原理解析

堆排序是一个十分高效的排序方法，其时间复杂度为：建堆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; //左孩子
        while(j<=n){
            //寻找第i个节点的最大的孩子
            if(j+1<=n && nums[j+1]>nums[j]){
                j=j+1;
            }

            //如果最大的孩子比父节点i还要大，则交换父子节点
            if(nums[i]<nums[j]){
                swap(nums[i],nums[j]);
                //然后再递归判断子节点
                i=j;
                j=2*i+1;
            }
            else{
                //此时父节点的值已经比最大的子节点都大了，说明满足大根堆的条件了，不用再调整了，返回
                return;
            }
        }

    }

建堆

从第一个非叶子节点出发，从右至左，从上至下依次调整堆。

void heapfy(vector<int> &nums){
        //从第一个非叶子节点出发，从右至左，从上至下依次调整堆
        //下标从0开始有n个元素的数组，第一个非叶子节点的下标为(n-1)/2
        int n=nums.size()-1;
        for(int i=(n-1)/2;i>=0;i--){
            adjust(nums,i,n);
        }
    }

堆排序函数

先建堆，再依次交换堆顶和末尾元素，再从堆顶元素开始调整堆。

vector<int> heapArray(vector<int>& nums) {
        int n=nums.size()-1;
        heapfy(nums);
        for(int i=n;i>=1;i--){
            swap(nums[0],nums[i]);
            adjust(nums,0,i-1);
        }
        return nums;
    }