关闭

查找最小的k个元素

标签: 最大堆最小k个数
720人阅读 评论(0) 收藏 举报
分类:

题目:输入n个整数,输出其中最小的k个。例如:输入1,2,3,4,5,6,7,8,则最小的4个数是1,2,3,4。

分析:采取一般的排序算法,如快速排序,堆排序,归并排序,等,先将这n个整数进行排序,然后再取最小的k个。那么时间复杂度为nlgn。

           我们可以从数组中取k个元素,建一个k个元素的最大堆,继续从数组里面取出元素,如果该元素大于堆中最大元素,那么舍弃,继续;如果小于,则将该元素替换掉堆中最大元素,并调整最大堆结构,维护最大堆性质不变,维护最大堆的性质,时间复杂度为lgk,那么总的时间复杂度为nlgk。这样最终堆中的k个元素即为最小的k个数。

           而STL中,set和multiset很好地实现了堆的性质,但是set中不允许存在相同的元素,而multiset则允许,那么我们采用multiset来实现最大堆。

#include "stdafx.h"
#include <vector>
#include <set>
#include <iostream>

using namespace std;

typedef multiset<int,greater<int>> IntSet;
void FindLeastKNumbers_(int a[], IntSet &leastNumbers, int nLength,int k)
{
	//a.clear();
	//如果原数组元素个数小于k,则不合题意,返回
	if(nLength < k)
		return;
	for(int i=0;i<nLength;i++)
	{
		if(leastNumbers.size()<k)
			leastNumbers.insert(a[i]);	//取前面k个元素建最大堆
		else
		{
			IntSet::iterator iterFirst = leastNumbers.begin();
			if(*iterFirst>(a[i]))
			{
				leastNumbers.erase(iterFirst);
				leastNumbers.insert(a[i]);
			}
		}
	}
}
int _tmain(int argc, _TCHAR* argv[])
{
	int a[20] = {1,2,7,9,3,6,2,9,8,10,11,17,12,18,19,16,17,14,5,20};
	IntSet heapset;
	FindLeastKNumbers_(a,heapset,20,6);
	IntSet::iterator iter = heapset.begin();
	for(;iter!=heapset.end();iter++)
		cout<<*iter<<",";
	return 0;
}

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:84485次
    • 积分:1136
    • 等级:
    • 排名:千里之外
    • 原创:31篇
    • 转载:22篇
    • 译文:1篇
    • 评论:25条
    最新评论