查找最小的k个元素


题目为输入n个元素找出这n个元素中最小的k个元素:如n=8,元素依次为1 2 3 4 5 6 7 8 则小的4的元素为1 2 3 4

下面上代码,解决方法见解题报告

/*
解题报告:
1、最简单的办法,给所有数据按升序或降序排序,从中取出相应的k个就可以了,最快的时间复杂度为nlog(n)
2、可以考虑维护一个堆,这个堆的中元素的个数为k,然后遍历元素,这时有两种情况:
	1、堆中元素的个数小于k,这时直接插入即可
	2、元素的个数为k,这时如果遍历到的元素小于堆中最大的那个元素就将堆中的这个元素删除,将遍历到的元素插入即可 
*/

#include <iostream>
#include <set>
#include <vector>
using namespace std;

typedef multiset<int, greater<int> > intHeap;

void FindNumbers(vector<int> &data, int k, intHeap &numbers)
{
	numbers.clear();
	if (0==k || data.size()<k)
		return ;
	vector<int>::const_iterator iter = data.begin();
	for (; iter!=data.end(); iter++)
	{
		if (numbers.size() < k)
		{
			numbers.insert(*iter);
		}
		else
		{
			intHeap::iterator heapIter = numbers.begin();
			if (*iter < *(numbers.begin()))
			{
				numbers.erase(heapIter);
				numbers.insert(*iter);
			}
		}
	}
}

int main()
{
	int n;
	vector<int> data;
	intHeap numbers;
	while (cin>>n)
	{
		for (int i=0; i<n; ++i)
		{
			int nTemp;
			cin>>nTemp;
			data.push_back(nTemp);
		}
		FindNumbers(data, 4, numbers);
		intHeap::const_iterator iter = numbers.begin();
		for (; iter!=numbers.end(); ++iter)
		{
			cout<<*iter<<" ";
		}
		cout<<endl;
	}
	return 0;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
您好,关于Python实现查找最小的k个数,我可以提供以下两种常见的方法: 方法一:使用堆 将数组前k个元素建立小根堆,然后遍历剩余元素,若比堆顶元素小,则将其加入堆中,同时将堆顶元素删除,最终堆中的k个元素即为所求。具体代码如下: import heapq def find_k_min(nums, k): if not nums or k <= 0 or k > len(nums): return None res = [] for i in range(k): heapq.heappush(res, nums[i]) for i in range(k, len(nums)): if nums[i] < res[0]: heapq.heappop(res) heapq.heappush(res, nums[i]) return res 方法二:使用快排的思想 使用快排的思想,每次选择一个基准元素,将小于基准元素的放在左边,大于等于基准元素的放在右边,然后判断基准元素的位置,若该位置等于k-1,则基准元素以及左边的元素就是所求;若该位置小于k-1,则在右半边继续查找;若该位置大于k-1,则在左半边继续查找。具体代码如下: def find_k_min(nums, k): if not nums or k <= 0 or k > len(nums): return None left, right = 0, len(nums)-1 while True: index = partition(nums, left, right) if index == k-1: return nums[:k] elif index < k-1: left = index + 1 else: right = index - 1 def partition(nums, left, right): pivot = nums[left] while left < right: while left < right and nums[right] >= pivot: right -= 1 nums[left] = nums[right] while left < right and nums[left] < pivot: left += 1 nums[right] = nums[left] nums[left] = pivot return left 希望能对您有所帮助!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值