题目描述
输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。
import java.util.*;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<>();
if(k>input.length){
return list;
}
PriorityQueue<Integer> pq = new PriorityQueue<>();
//构造小顶堆,O(n)
for(int i=0; i<input.length; i++){
pq.offer(input[i]);
}
//取最小的k个
for(int i=0; i<k; i++){
list.add(pq.poll());
}
return list;
}
}
2017.6.2 第二次做，用了排序，时间复杂度O(n^2)比用小顶堆要差
import java.util.*;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<>();
if(input.length < k){
return list;
}
Arrays.sort(input);
for(int i=0; i<k; i++){
list.add(input[i]);
}
return list;
}
}
