数组中最小的k个数
题目:输入整数数组 arr ,找出其中最小的 k 个数。例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
思路:使用快排的思想,我们知道,快排的一次划分过程的结果是将所有小于基准元素的元素放在该基准元素的前面,所有大于基准元素的元素放在该基准元素的后面。这样,我们可以使用快排的思想。
① 假设第一划分得到的划分左右边界的下标值为j
,如果k==j
,则说明 最小的k
个数就在j
的左边,返回0~j-1
这个区间的数据。
② 如果k<j
,说明最小的k个数存在右边界( 要注意区分和①的不同,如果k<j,那么只能说明最小的k个数存在于右边界,不能说明0~k-1就是最小的k个数,比如这种情况:[1,5,3,7,15,12],该数组是第一次划分后得到的,我们现在找最小的2个数,此时 k==2
、j==3
,k<j
,我们不能说0~2
这个区间的元素就是我们要问题的答案,这样的结果就成了[1,5,3]
,很明显答案不对 ) 则我们在右边界进行快速排序,即把右边界一分为二。
③ 如果k>j
,则在右边界继续划分。注意,此时的k的取值问题。
与快排的思想不同的是,快排需要左右边界同时划分,而我们仅仅是往一侧划分即可。
代码实现:
public int[] getLeastNumbers(int[] arr, int k) {
if(arr==null || arr.length == 0) return new int[0];
if(k<=0) return new int[0];
int nums [] =new int[k];
search(arr,0,arr.length-1,k);
for (int i = 0; i < k; i++) {
nums[i] = arr[i];
}
return nums;
}
private void search(int[] arr,int left,int right,int k){
int pos = partition(arr,left,right);
// num表示左侧有多少个元素。
int num = pos-left+1;
if (k==num){
return ;
}else if(k > num){
// 如果 k > num 则说明 左半段的元素已经是必须要输出的了。
// 继续从后半段查找 剩下的满足条件的 即k-num个元素。
search(arr,pos+1,right,k-num);
}else{
search(arr,left,pos-1,k);
}
}
// 快排的一次划分
private int partition(int[] arr,int left,int right){
int i = left;
int j = right;
int tag = arr[i];
while(i<j){
while(i<j && arr[j] >= tag) j--;
if (i<j) arr[i] = arr[j];
while(i<j && arr[i] <= tag) i++;
if (i<j) arr[j] = arr[i];
}
arr[i] = tag;
return i;
}