面试题40:最小的K个数

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

思路：使用set，因为set的底层是红黑树实现的，红黑树是有序结构，处理插入删除非常方便。

假设set的大小为k，那么当set的大小小于k时，直接往set里面插入那个数，log(k);

那么当set的大小大于k时，我们只需要比较set中最大的数，和当前的数，删除较大数，插入较小数就好,log(k);

那么如何取得当前集合中的最大数呢？

multiset<int,greater<int> >s;//从大到小排列

这里，s.begin()就是最大的了。

总复杂度是nlog(k).

//使用优先队列的复杂度也是nlog(k),不过优先队列的底层是堆实现。

代码：

#include <iostream>
using namespace std;
#include <vector>
#include <set>
#include <queue>
#include <algorithm>
#include <functional>
class Solution {
public:
	vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
		if (input.size() < k || k <= 0) 
			return vector<int>();
		multiset<int,greater<int> >s;//从大到小排列
		for (vector<int>::iterator it = input.begin(); it != input.end(); it++)
		{
			if (s.size() < k)
				s.insert(*it);
			else
			{
				//multiset<int, greater<int>>::iterator it1 = s.begin();
				if (*it < (*s.begin()))
				{
					s.erase(*s.begin());
					s.insert(*it);
				}
			}
		}
		return vector<int>(s.begin(), s.end());
		
	}
};
int main()
{
	Solution s;
	int num1, num2;
	vector<int> a = { 4, 5, 1, 6, 2, 7, 3, 8 };
	vector<int> ans = s.GetLeastNumbers_Solution(a, 4);
	for each (int num in ans)
	{
		cout << num << endl;
	}
	return 0;
}