题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
第一思路 先排序然后找出前k个数 这种算法复杂度和排序的时间复杂度一样 最优为O(nlogn) 下面介绍一种方法 时间复杂度为O(nlogk);
解题思路
1.维护一个大根堆 遍历数组先放k个数进堆 堆顶的元素就是这个k个元素中最大的
2.继续遍历 如果当前数大于堆顶元素 删除堆顶元素 当前元素入堆
遍历完数组后堆里的元素 就是k个最小的。
因为只需要维护k个元素的堆 所以该算法时间复杂度为0(nlogk);
下面是代码实现
import java.util.PriorityQueue;
import java.util.Comparator;
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList();
if (k > input.length || k == 0) {
return list;
}
PriorityQueue<Integer> queue = new PriorityQueue(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2 - o1;
}
});
for (int i = 0; i < input.length; i++) {
if (i < k) {
queue.offer(input[i]);
}else if(queue.peek()>input[i]) {
queue.poll();
queue.offer(input[i]);
}
}
for (int i = 0; i < k; i++) {
list.add(queue.poll());
}
return list;
}
}