剑指offer 最小的K个数

题目描述

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

 

 

 

思路一：直接排序，时间复杂度O(nlogn)

思路二：建立最大堆。

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        int len = input.size();
        if (k <= 0 || k > len) //k不符合条件的话，均返回空
            return res;
        priority_queue<int,vector<int> > pq;
        for (int i = 0; i < len; i++) {
            if (pq.size() < k) {
                pq.push(input[i]);
            } else {
                int temp = pq.top();
                if (input[i] < temp) {
                    pq.pop();
                    pq.push(input[i]);
                }
            }
        }
        while (!pq.empty()) {
            int top = pq.top();
            pq.pop();
            res.push_back(top);
        }
        //sort(res.begin(), res.end());
        return res;
    }
};

 

思路三：快排思想