问题: 求一维数组中的最小的k个数。
思路1: 排序。把数组从小到大排序,取前k个数。若数组过大,机器内存无法同时容纳整个数组,则需要使用外部排序。
思路2: 堆。创建一个最小堆,初始化大小为k;堆顶为堆的最大元素。扫描一遍数组得到k个最小元素。
思路3: 快排分区。对数组调用分区函数,如果返回的下标是k-1,那么数组左边的k个数就是结果。
思路3 java代码:
int partition(int[] data,int start,int end){
if(start>end)
return -1;
int index=start;
int tmp=data[index];
data[index]=data[end];
for(int i=start;i<end;i++){
if(data[i]<=data[end]){
if(i!=index){
tmp=data[index];
data[index]=data[i];
data[i]=tmp;
}
index++;
}
}
tmp=data[end];
data[end]=data[index];
data[index]=tmp;
return index;
}
void getTopK(int[] data,int k){
int start=0,end=data.length-1;
int index=partition(data,start,end);
while(index!=k-1){
if(index>k-1){
end=index-1;
index=partition(data,start,end);
}
else{
start=index+1;
index=partition(data,start,end);
}
}
for(int i=0;i<k;i++){
System.out.println(data[i]+"\t");
}
}