剑指offer 面试题40 最小的K个数

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

输入：数组，K

输出：最小的K个数

思路：

思路一（可以修改数组）基于Partition函数

思路二 基于priority_queue（大顶堆）时间复杂度为

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        priority_queue<int> bigHeap;
        int i=0;
        int len=input.size();
        if(k==0||len<k)
            return {};
        if(len==k)
            return input;
        for(;i<k;++i)
            bigHeap.push(input[i]);
        for(;i<len;++i)
        {
            if(input[i]<bigHeap.top())
            {
                bigHeap.pop();
                bigHeap.push(input[i]);
            }
        }
        vector<int> res(k);
        for(int j=0;j<k;j++)
        {
            res[j]=bigHeap.top();
            bigHeap.pop();
        }
        return res;
    }
};