题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路
本题关键就是对数组排序,排序算法决定本题的时间复杂度
法1
最简单的思路,利用Java自带的sort方法(快速排序),时间复杂度O(nlogn)
法2
被广泛认同的解法:使用堆排序。若使用大顶堆,时间复杂度O(logk)
求最小的k个数,就建k个大顶堆,每个堆的堆顶元素始终是最小的(若有更小的元素进来,就把当前堆顶删掉,加入新的元素作为新堆顶)
最终输出k个堆顶,即为数组中的k个最小的数
使用优先队列实现堆,默认创建的是小顶堆,为了得到大顶堆,需要使用比较器comparator接口进行反转,再用lambda表达式传入比较器是比较简洁的写法(Java自带的方法reverseOrder也可以,但用的时候想不起来)
实现
法1:快排(Java自带方法,一行实现)
import java.util.Arrays;
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> res=new ArrayList<>();
//最简单的思路:排序后输出前k个(Java的sort方法是快速排序)
//特殊情况
if(input.length==0 || k<=0 || k>input.length) return res;
Arrays.sort(input);
for(int i=0;i<k;i++){
res.add(input[i]);
}
return res;
}
}
法2:堆排序,大顶堆,时间复杂度较小
import java.util.PriorityQueue;
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> res=new ArrayList<>();
//堆排序:大顶堆排序时间复杂度更低(用优先队列实现堆)
//特殊情况
if(input.length==0 || k<=0 || k>input.length) return res;
//创建大顶堆(优先队列默认创建小顶堆,所以要使用comparator接口进行反转:用lambda实现)
//lambda表达式,传入一个比较器,规则是大的反而小,所以得到一个大顶堆
PriorityQueue<Integer> maxHeap=new PriorityQueue<>((a,b)->b - a);
for(int i=0;i<input.length;i++){
//如果大顶堆中没有k个堆,继续加入元素,凑够k个堆
if(maxHeap.size()<k){
maxHeap.offer(input[i]);
}
else{ //若当前遍历的元素比堆顶小,就删掉当前堆顶,把当前元素加入变成新的堆顶
if(input[i]<maxHeap.peek()){
maxHeap.remove(); //删除当前堆顶
maxHeap.offer(input[i]);
}
}
}
//只要大顶堆里还有元素,就全取出来加到结果集合中
while(!maxHeap.isEmpty()){
res.add(maxHeap.remove());
}
return res;
}
}