題目描述
輸入n個整數(shù)，找出其中最小的K個數(shù)悼潭。例如輸入4,5,1,6,2,7,3,8這8個數(shù)字庇忌，則最小的4個數(shù)字是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<>();
//構(gòu)造小頂堆,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 第二次做舰褪，用了排序皆疹，時間復(fù)雜度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;
}
}
