题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
tips: 使用C++ STL 中的multiset,维护一个O(k)大小的红黑树,时间复杂度O(nlog(n))
学习了set, multiset的基本操作以及内部原理(红黑树)
typedef multiset<int,greater<int> > IntSet;
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> res;
if(k<=0||input.size()<k) {
return res;
}
IntSet myset;
for (int i = 0; i < input.size(); i++)
{
if(myset.size()<k) {
myset.insert(input[i]);
} else {
auto iter=myset.begin();
if(input[i]<(*iter)) {
myset.erase((*iter));
myset.insert(input[i]);
}
}
}
for (auto iter = myset.end(); iter!= myset.begin(); iter--)
{
res.push_back((*iter));
}
return res;
}
};