《剑指offer》最小的K个数

最新推荐文章于 2022-03-06 10:10:36 发布

原创最新推荐文章于 2022-03-06 10:10:36 发布 · 806 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer #牛客网

《剑指offer》题目解析同时被 2 个专栏收录

66 篇文章

订阅专栏

《剑指offer》专题

65 篇文章

订阅专栏

本文介绍了一种高效算法，用于从一组整数中找到最小的K个数。通过部分排序而非完全排序的方法来提高效率，特别适用于K相对较小的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：http://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf?rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

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

思路
我们可以直接排序然后取出前K个数，当然这种时间复杂度并不是最好的，对于k比较小的时候，直接排序会比较浪费时间
我们可以先保存前K个数进结果数组，然后再将其排序，然后当数组出现新的比现在结果数组最大的数小的数的时候，我们再弹出结果数组最大的数

class Solution
{
	public:
		vector<int> GetLeastNumbers_Solution(vector<int> input, int k)
		{
			int len = input.size();
			vector<int> ans;
			if(k<1 || k>len)
				return ans;
			for(int i = 0; i<len; i++)
			{
				if(ans.size()<k)
				{
					ans.push_back(input[i]);
					if(ans.size()==k)
						sort(ans.begin(),ans.end());
				}
				else
				{
					if(input[i]<ans[k-1])
					{
						ans[k-1] = input[i];
						if(k>1 && ans[k-1]<ans[k-2])
							sort(ans.begin(),ans.end());
					}
				}
			}
			return ans;
		}
};