【牛客】最小的k个数

使用topK的方法，如果不想写堆的话，了解几个库里的堆的函数

std::make_heap将[start, end)范围进行堆排序，默认使用less, 即最大元素放在第一个。
std::pop_heap将front（即第一个最大元素）移动到end的前部，同时将剩下的元素重新构造成(堆排序)一个新的heap。
std::push_heap对刚插入的（尾部）元素做堆排序。
std::sort_heap将一个堆做排序,最终成为一个有序的系列，可以看到sort_heap时，必须先是一个堆（两个特性：1、最大元素在第一个 2、添加或者删除元素以对数时间），因此必须先做一次make_heap.

思路：
1、参数检验：len<=0、k>len、k<=0 都是不对的
2、建堆：使用make_heap建立k个节点的最大堆
3、使用剩余元素和堆顶元素比较，
4、如果小于堆顶元素，那么删除堆顶元素，将此元素插入
5、最后使用sort_heap排序，输出

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        //topK
        int len=input.size();
        if(len<=0||k>len||k<=0)
            return vector<int>();
        //建立k个元素的堆(大堆)
        vector<int> res;
        for(int i=0;i<k;i++){
            res.push_back(input[i]);
        }
        make_heap(res.begin(),res.end());
        for(int i=k;i<len;i++){
            if(res[0]>input[i]){
                pop_heap(res.begin(),res.end());//将待删除元素放到最后
                res.pop_back();
                res.push_back(input[i]);
                push_heap(res.begin(),res.end());
            }
        }
        //最后，再从小到大输出
        sort_heap(res.begin(),res.end());
        return res;
    }
};