最小的K个数

题目描述

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

 

 

 

提交链接：点击

 

 

 

思路：

　　方法一：利用c++的sort函数将n个整数排序，然后选出前K个数存入结果并返回，时间复杂度为O(nlogn)。

       方法二：利用大顶堆，首先建立前K个元素的大根堆，然后遍历后K个元素，如果比堆顶元素小，则入堆，重新生成大根堆；如果比堆顶元素大，则不进行操作；

　　　　　　最后得到前K个小的数在vector中，利用sort函数排序输出最后结果即可，时间复杂度为O(nlogk)。

 

 

代码：

//第一种方法
class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> result;
        if(k>input.size()) return result;
        sort(input.begin(),input.end());
        for(int i=0;i<k;i++){
            result.push_back(input[i]);
        }
        return result;
    }
};

//方法二
class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len=input.size();
        if(len<=0 || k<=0 || k>len) return vector<int>();
        vector<int> result(input.begin(),input.begin()+k);
        make_heap(result.begin(),result.end());
        for(int i=k;i<len;i++){
            if(input[i]<result.front()){  //比大顶堆值小，说明要将新元素插入堆
                pop_heap(result.begin(),result.end());  
                result.pop_back();
                
                result.push_back(input[i]);
                push_heap(result.begin(),result.end());
            }
        }
        sort(result.begin(),result.end());
        return result;
    }
};