29、最小的k个数

题目描述
输入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;
    }
}