题目:
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
分析:
传统的做法可以是o(nlog(n)),就是先排序然后从前面中取出k个值。高效的做法可以实现o(nlogk),
借助一个容器,容量k,将数组的读入k个数存放在容器中,然后继续访问数组后面的数,将其后每个数和容器中最大值比较,
比最大值小则剔除容器中最大值,同时将该数放入容器!这样就可以保证遍历整个数组之后,容器中的数就是要找的k个较小数。
这种算法策略非常适合大数据,并且可以不必要将整个数组加载入内存!容器要支持具有排序功能,选择C++的multiset!
#include<vector>
#include<iostream>
#include<set>
#include<algorithm>
#include<functional>
using namespace std;
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
int len = input.size();
if (len <= 0 || k>len)
return vector<int>();
multiset<int, greater<int>> leastNums;//红黑树二叉树,降序(最大堆)
vector<int>::iterator vec_it = input.begin();
for (; vec_it != input.end(); vec_it++)
{
if (leastNums.size()<k)//填入容器k个值
leastNums.insert(*vec_it);
else
{
multiset<int, greater<int>>::iterator greast_it = leastNums.begin();//每一种容器就需要特定的迭代器
if (*vec_it<*greast_it)
{
leastNums.erase(greast_it);//弹出容器最大值
leastNums.insert(*vec_it);//将当前数组值插入
}
}
}
return vector<int>(leastNums.begin(), leastNums.end());
}
int main()
{
vector<int> input = {2,4,5,7,89,1,0};
vector<int> least_k=GetLeastNumbers_Solution(input,5);
vector<int>::iterator it = least_k.begin();
for (; it != least_k.end(); it++)
{
cout << *it<<endl;
}
while (true)
{
}
return 0;
}