博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
程序员面试题100题第05题——查找最小的K个元素
阅读量:6835 次
发布时间:2019-06-26

本文共 2125 字,大约阅读时间需要 7 分钟。

转自:

题目:输入n个整数,输出其中最小的k个。

例如输入1,2,3,4,5,6,7和8这8个数字,则最小的4个数字为1,2,3和4。

分析:这道题最简单的思路莫过于把输入的n个整数排序,这样排在最前面的k个数就是最小的k个数。只是这种思路的时间复杂度为O(nlogn)。我们试着寻找更快的解决思路。

我们可以开辟一个长度为k的数组。每次从输入的n个整数中读入一个数。如果数组中已经插入的元素少于k个,则将读入的整数直接放到数组中。否则长度为k的数组已经满了,不能再往数组里插入元素,只能替换了。如果读入的这个整数比数组中已有k个整数的最大值要小,则用读入的这个整数替换这个最大值;如果读入的整数比数组中已有k个整数的最大值还要大,则读入的这个整数不可能是最小的k个整数之一,抛弃这个整数。这种思路相当于只要排序k个整数,因此时间复杂可以降到O(n+nlogk)。通常情况下k要远小于n,所以这种办法要优于前面的思路。

这是我能够想出来的最快的解决方案。不过从给面试官留下更好印象的角度出发,我们可以进一步把代码写得更漂亮一些。从上面的分析,当长度为k的数组已经满了之后,如果需要替换,每次替换的都是数组中的最大值。在常用的数据结构中,能够在O(1)时间里得到最大值的数据结构为最大堆。因此我们可以用堆(heap)来代替数组。

另外,自己重头开始写一个最大堆需要一定量的代码。我们现在不需要重新去发明车轮,因为前人早就发明出来了。同样,STL中的set和multiset为我们做了很好的堆的实现,我们可以拿过来用。既偷了懒,又给面试官留下熟悉STL的好印象,何乐而不为之?

#include 
#include
#include
using namespace std;typedef multiset
> IntHeap; //最大堆 // 这里必须使用空格隔开两个相邻的>号,否则编译器会认为是输入>>符号///// find k least numbers in a vector///void FindKLeastNumbers( const vector
& data, // a vector of data IntHeap& leastNumbers, // k least numbers, output unsigned int k ){ leastNumbers.clear(); if(k == 0 || data.size() < k) return; vector
::const_iterator iter = data.begin(); for(; iter != data.end(); ++ iter) { // if less than k numbers was inserted into leastNumbers if((leastNumbers.size()) < k) leastNumbers.insert(*iter); // leastNumbers contains k numbers and it's full now else { // first number in leastNumbers is the greatest one IntHeap::iterator iterFirst = leastNumbers.begin(); // if is less than the previous greatest number if(*iter < *(leastNumbers.begin())) { // replace the previous greatest number leastNumbers.erase(iterFirst); leastNumbers.insert(*iter); } } }}int main(){ vector
svec; for (int i = 0;i < 10; i ++) { svec.push_back(i); } IntHeap leastNumbers; FindKLeastNumbers(svec,leastNumbers,4); IntHeap::iterator iter = leastNumbers.begin(); for (; iter != leastNumbers.end(); iter ++) { cout<<*iter<<'/t'; } cout<

 

 

转载于:https://www.cnblogs.com/zjhnl/archive/2012/10/02/2710560.html

你可能感兴趣的文章
SysUtils 中其他 PChar 函数: StrPas、StrNew、StrDispose、StrNextChar、StrFmt、StrLFmt、StrCharLength...
查看>>
对数据进行处理
查看>>
Delphi 的绘图功能[14] - 图像类
查看>>
常用函数总结
查看>>
map/reduce执行流程
查看>>
CentOS 7 安装Nginx-1.9.2
查看>>
pythonGUI -- pyside安装与初试
查看>>
我的友情链接
查看>>
写给大学时代的我们——英语很重要
查看>>
Access模糊查询
查看>>
redmine安装部署文档
查看>>
Exchange 2010做OWA的https重定向时会话延迟
查看>>
Spring与Quartz实现定期任务
查看>>
设计模式之建造者模式实例
查看>>
我的友情链接
查看>>
我的友情链接
查看>>
用Java写算法之五:快速排序
查看>>
UK 更新惊魂记
查看>>
ZooKeeper-3.3.4集群安装配置
查看>>
《Spring In Action》第三版中文版 Chapter 1 Piece 3
查看>>