本文最后更新于:2020年6月29日 晚上

题目描述:

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

实现如下:

//测试用例中给出的返回值是有序的。。啧啧。。
//找到序列中最小的k个数,利用set,对数字进行insert或erase
//红黑树中查找、插入、删除操作都为 O(logk)
//最终将set中存储的值返回值vector中
//方法不改变原vector
//注意:
//1.判断传入的vector是否为空
//2.判断传入的k是否合法
class Solution 
{
public:
	vector<int> GetLeastNumbers_Solution(vector<int> input, int k) 
	{
		vector<int> output;
		if (input.size() <= 0 || k <= 0 || k > input.size()) return output;//防御性动作
		set<int> se;
		set<int>::iterator it;
		for (int i = 0; i < input.size(); ++i)//遍历原vector中的所有数
		{
			if (i < k)//如果set中的个数小于k,则直接插入
				se.insert(input[i]);
			else//否则将set中的最大值和vector中第i个数比较
			{
				it = --se.end();//取set中的最大值
				if (*it > input[i])//若set中的最大值大于vector中第i个数
				{
					se.erase(it);//删除set中的最大值
					se.insert(input[i]);//插入vector中第i个数
				}
			}
		}
		
		it = se.begin();
		for (; it != se.end(); ++it)//将set中的数以vector形式return
		{
			output.push_back(*it);
		}
		return output;
	}
};