最小的K个数

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

思路1：冒泡排序O(n*k)
当然，如果是海量数据，那么此时效率太低

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        if(k > input.size() || k <= 0)
            return res;
        for(int i = input.size();i > input.size() - k;i--){
            int min = 0;
            for(int j = 0;j < i;j++){
                if(input[min] < input[j]){
                    swap(input[min],input[j]);
                }
                min = j;
            }
            res.push_back(input[min]);
        }
        return res;
    }
};

思路2：堆排序O(nlogk)
先把这n个数中的前k个输入进大顶堆
然后遍历后面的数，
如果比堆顶小，就删除堆顶，并把这个数入堆
否则继续

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