输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路描述:通过快速排序思想,排完后,位于左边全是小于关键字,右边全是大于关键字,若将关键字的下标等于k-1,那么左边结果即是所求,需要注意的是,左边并不一定是排好序的。
import java.util.List;
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
return getKthLeastNumber(input,k);
}
private static ArrayList<Integer> getKthLeastNumber(int[] a, int k) {
ArrayList<Integer> list=new ArrayList<Integer>();
if(a==null||a.length==0||k<=0||k>a.length)return list;
int index=quickSort(a,0,a.length-1);
int start=0;
int end=0;
while(index!=k-1){
if(index>k-1){
end=index-1;
index=quickSort(a,0,end);
}
else{
start=index+1;
index=quickSort(a,start,end);
}
}
for(int i=0;i<k;i++)
list.add(a[i]);
return list;
}
private static int quickSort(int[] a, int left, int right) {
if(left>=right)return left;
int i=left;
int j=right;
int key=a[left];
while(i<j){
while(i<j&&a[j]>=key)j--;
a[i]=a[j];
while(i<j&&a[i]<=key)i++;
a[j]=a[i];
}
a[i]=key;
quickSort(a,left,i-1);
quickSort(a,i+1,right);
return i;
}
}