标签:实现 技术 最大 erase clu bubuko alt cout 数字
输入n个整数,找出其中最小的k个数。例如输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
1.于快速排序中的Partition函数来解决这个问题。如果基于数组的第k个数字来调整,使得比第k个数字小的所有数字都位于数组的左边,比第k个数字大的所有数字都位于数组的右边。这样调整之后,位于数组中左边的k个数字就是最小的k个数字(这k个数字不一定是排序的),采用这种思路是有限制的。我们需要修改输入的数组,因为函数Partition会调整数组中数字的顺序
2.可以先创建一个大小为k的数据容器来存储最小的k个数字,接下来我们每次从输入的n个整数中读入一个数。
因此当容器满了之后,我们要做3件事情:一是在k个整数中找到最大数;二是有可能在这个容器中删除最大数;三是有可能要插入一个新的数字。如果用一个二叉树来实现这个数据容器,那么我们能在O(logk)时间内实现这三步操作。因此对于n个输入数字而言,总的时间效率就是O(nlogk)
#include <iostream> #include <set> #include <vector> #include <algorithm> using namespace std; class Solution { public: void get_least_num(const vector<int> &v,multiset<int,greater<int> > &least,int k); }; void Solution::get_least_num(const vector<int> &v,multiset<int,greater<int> > &least,int k) { if(k>=v.size()||k<=0||v.empty()) return; least.clear(); for(auto it=v.begin();it!=v.end();++it) { if(least.size()<k) least.insert(*it); else { if((*it)<(*least.begin())) { least.erase(*least.begin()); least.insert(*it); } } } } int main() { Solution s; vector<int> v{9,8,1,2,3,4,5}; multiset<int,greater<int> > m; s.get_least_num(v,m,3); for(auto k:m) cout<<k<<" "; cout<<endl; return 0; }
假设题目是要求从海量的数据中找出最小的k个数字,由于内存的大小是有限的,有可能不能把这些海量的数据一次性全部载入内存。这个时候,我们可以从辅助存储空间(比如硬盘)中每次读入一个数字,根据GetLeastNumbers的方式判断是不是需要放入容器leastNumbers即可。
这种思路只要求内存能够容纳leastNumbers即可,因此它最适合的情形就是n很大并且k较小的问题。
标签:实现 技术 最大 erase clu bubuko alt cout 数字
原文地址:https://www.cnblogs.com/tianzeng/p/10230671.html