试编写一个算法,使之能够在数组L[1…n]中找出第k小的元素(即从小到大排序后处于第k个位置的元素)
重要的事情说三遍:代码不是我写的!代码不是我写的!代码不是我写的!
这是王道数据结构第8章第3节的一道题,记录在博客上方便以后回顾。
如有谬误或者不足还请批评指正!
以下是书上对该题的解答:
本题最直接的做法是用排序算法对数组先进行从小到大的排序,然后直接提取L(k)便得到了第k小元素,但其平均时间复杂度将达到O(nlog2n)以上。此外,还可采用小顶堆的方法,每次堆顶元素都是最小值元素,时间复杂度为O(n + klog2n)。下面介绍一个更精彩的算法,它基于快速排序的划分操作。
这个算法的主要思想如下:从数组L[1…n]中选择枢轴pivot(随机或直接取第一个)进行和快速排序一样的划分操作后,表L[1…n]被划分为L[1…m-1]和L[m+1…n],其中L(m)=pivot。
讨论m与k的大小关系:
1)当m=k时,显然pivot就是所要寻找的元素,直接返回pivot即可。
2)当m<k时,所要寻找的元素一定落在L[m+1…n]中,因此可对L[m+1…n]递归地查找第k-m小的元素。
3)当m>k时,所要寻找的元素一定落在L[1…m-1]中,因此可对L[1…m-1]递归地查找第k小的元素。
该算法的时间复杂度在平均情况下可以达到O(n),而所占空间的复杂度则取决于划分的方法。
算法的实现如下: