日常生活中我们总是遇到求解一个数组的最小值或者最大值等问题,这些问题基本上都可以在线性时间复杂度内完成,但是如果需要寻找数组的前K个最小值时该怎么做呢?最一般的想法就是将数组排序,然后取出排序后的数组的前k个元素即可,这种算法的时间复杂度也就是排序的时间复杂度为O(NlogN)。但是仔细一想发现,我们没有必要对所有的数进行排序,因为我们只用到了排序后的前K个数,为此可以考虑只找出排序的前K个数即可。这就让我们联想到了堆排序的过程。
堆排序分为两部分,一部分是初始化建堆,另一部分是堆的调整,下面以建立小顶堆为例说明一下堆排序的过程。
- 建堆
- 调整堆
堆建完之后,堆顶的元素即为数组的最小值,然后将其取出,将最后一位叶子元素放到堆顶,接着继续调整堆,然后再取出堆顶元素,如此重复上述步骤K次,即可得到数组中前K个最小元素。
本文以K=2为例子实现了上述算法
建堆:
void Create_Heap(int* a, const int n)
{
for(int i=(n-1)/2; i>=0; i--)
{
Exchange_Node(a,n,i);
}
}
调整堆:
void