【剑指offer】最小的K个数

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

解：使用堆这一数据结构，时间复杂度为O(NlgK)

class Solution {public:    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {         // input为空，input元素个数少于k，k非正数，则直接返回        if(input.size() == 0 || input.size() < k || k <= 0)             return vector<int>();        vector<int> heap(input.begin(), input.begin() + k);        // 建堆        // 默认是最大堆，创建最小堆需添加参数greater<int>()        make_heap(heap.begin(), heap.end());        for(auto it = input.begin() + k; it != input.end(); ++it){            // 最大堆的最大元素就在索引为0的位置            if(*it < heap[0]){                //先pop_heap,然后在容器中pop_back删除                pop_heap(heap.begin(), heap.end());                heap.pop_back();                //先在容器中push_back加入，再push_heap                heap.push_back(*it);                push_heap(heap.begin(), heap.end());            }        }        // 从小到大输出, k一般比较小，直接sort即可        sort_heap(heap.begin(), heap.end());             return heap;             }};