【剑指Offer】最小的K个数 两种简单解法

题目描述

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

第一种:

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

第二种：

利用堆排序，O(Nlogk)，适合处理海量数据

(1) 遍历输入数组，将前k个数插入堆中，利用mutilset来实现

(2)继续从输入数组中读入元素作为待插入元素，将它与堆中最大元素比较。如果比堆中的元素小，则进行替换。直到输入数组遍历结束。

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res;
        if(input.empty()|| k<0 ||input.size()<k)
            return res;
        multiset<int,greater<int>> leastNumbers;//从大到小排序
        multiset<int,greater<int>>::iterator itermin;//迭代器
        
        vector<int>::iterator iter=input.begin();
        for(;iter!=input.end();iter++){
            //先将前k个数加入堆中
            if(leastNumbers.size()<k)
                leastNumbers.insert(*iter);
            else{
                //因为是最大堆，从大到小排序，第一个为最大
                itermin = leastNumbers.begin();
                //若当前的值比最大堆中的小则删除大的，插入小的
                if(*iter < *(leastNumbers.begin())){    
                    leastNumbers.erase(itermin);
                    leastNumbers.insert(*iter);
                }
            }
        }
       for(itermin=leastNumbers.begin();itermin!=leastNumbers.end();itermin++)
           res.push_back(*itermin);
        
        return res;
    }
};