剑指offer | 最小的K个数

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

题目解析：首先想到的就是插入法，。改进一下就是前面一直保持7个有序数字。遍历剩下数组，元素若小于最大值就交换。用C++的set来解决代码就简洁多了，删除插入都是logk。

 

代码如下：

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        set<int> kset;
        if(k <= 0 || input.size() < unsigned(k)) return vector<int>(0);
        vector<int>::iterator iter = input.begin(), iend = input.end();
        while(k--) {
            kset.insert(*iter++);
        }
        
        while(iter != iend) {
            if(*iter < *kset.rbegin()) {
                kset.erase(*kset.rbegin());
                kset.insert(*iter++);
            }
            else ++iter;
        }
        
        vector<int> ans(kset.begin(), kset.end());
        return ans;
    }
};

拓展：quick select