问题
在数组S中查找第k小的元素。
解析
k=|S1|+1,m就是所要找的第k小的数;(以m为划分标准后,比m小的有|S1|个,如果恰巧,k=|S1|+1,则m就是所要找到第k小的数
k<=|S1|,归约为在S1中找第k1小的子问题,k1在子问题中的相对位置不变,即k1=k;
k>|S1|+1,归约为在S2中找k2位置的子问题,k2相对于S2子问题和k相对于S的关系,即k2=k-|S1|-1。(在S中找k,就是在S2中找k2)
设计
Select(S,k)
输入:n个数的数组S,正整数k
输出:S中的 第k小元素
将S划分成5个一组,共n/5个组
每组找一个中位数,把这些中位数放到集合M中
34m←Select(M,|M|/2) //选 M的中位数m ,将S中的数划分成A,B,C,D四个集合
把A和D中的每个元素与m比较,小的构成S1,大的构成S2 ;
S1<- S1 U C S2<- S2 U B;
if k=|S1|+1 then 输出m
else if k≤|S1|
then Select(S1, k)
else Select(S2,k-|S1|-1)
分析
W(n)=O(n)
源码
https://github.com/yaoshuangice/yaoshuang