最小的K个数

题目描述
输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。

思路：
直接排序来做时间复杂度是nlog(n)
考虑开辟一个额外的空间，空间大小为k，在遍历数小于等于k时，直接将其加到额外空间中；在遍历数大于等于k时，就要考虑额外空间中最大数是否大于当前遍历的数，若大于，则当前遍历的数替换它；若小于，则不做额外操作。
考虑到每一次都要从额外数组中获取最大值，显然使用大顶堆合适，其获取操作o(1)，插入删除操作o(logn)
因此总的时间复杂度nlog(k)

import java.util.ArrayList;
import java.util.PriorityQueue;
import java.util.Comparator;
import java.util.Iterator;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> res=new ArrayList<Integer>();
        int len=input.length;
        if(len<k||k<=0)return res;
        PriorityQueue<Integer> maxHeap=new PriorityQueue<Integer>(new Comparator<Integer>(){
            public int compare(Integer o1,Integer o2){
                return o2-o1;
            }
        });
        for(int i=0;i<len;i++){
            if(maxHeap.size()<k){
                maxHeap.add(input[i]);
            }else{
                int temp=maxHeap.peek();
                if(temp>input[i]){
                    maxHeap.poll();
                    maxHeap.add(input[i]);
                }
            }
        }
        Iterator it=maxHeap.iterator();
        while(it.hasNext()){
            res.add((Integer)it.next());
        }
        return res;
    }
}

快排思想o(n)

import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        int len=input.length;
        ArrayList<Integer> res=new ArrayList<>();
        if(k<=0||k>len)return res;
        getResult(input,k,0,len-1);
        for(int i=0;i<k;i++)res.add(input[i]);
        return res;
    }
    public void getResult(int[] input,int k,int left,int right){
        if(left>right)return;
        int target=input[left],begin=left,end=right;
        while(begin<end){
            while(begin<end&&input[end]>target)end--;
            if(begin<end)input[begin++]=input[end];
            while(begin<end&&input[begin]<target)begin++;
            if(begin<end)input[end--]=input[begin];
        }
        input[begin]=target;
        if(begin<k)getResult(input,k,begin+1,right);
        if(begin>k)getResult(input,k,left,begin-1);
        return;
    }
}