题目
编程之美上的寻找N个数中最大的前K数,给出了一种算法,我认为比较好:
算法是这样写的:
假设N个数存储在数组S中,我们从数组S中随机选出一个元素X,把数组分为两部分Sa和Sb.Sa中的元素都大于X,Sb中的元素都小于X,这时,有两种可能性:
1.Sa中元素的个数小于K,Sa中所有的数和Sb中最大的K-|Sa|个数(|Sa|指Sa中元素的个数)就是数组S中最大的K个数。
2.Sa中元素的个数大于或等于K,则直接返回Sa中最大的K个元素。
平均时间复杂度O(N*log2 K)
伪代码
伪代码如下:
Kbig(S,K)
if(k<=0):
return []
if(length S<=K):
return S
(Sa,Sb)=Partition(s)
return Kbig(Sa,k).Append(Kbig(Sb,k-length Sa)
//
Partition(S):
Sa=[];
Sb=[];
Swap(s[1],S[random()%length S])
p=S[1]
for i in [2:length S]:
s[i] > p ? Sa.Append(s[i]):Sb.Append(S[i])
length Sa
return (Sa,Sb)
实现
// MicroSTest2.cpp : Defines the entry point for the console application.
//
#include &