最小的k个数

来源:互联网 发布:linux shutdown命令 编辑:程序博客网 时间:2024/06/08 13:12

if 0

/*
面试题30:最小的k个数
:输入n个整数,找出其中最小的k个数,例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4.
解题思路:
1.在n个整数中找最小的k个数,我们可以创建一个大小为k的数据容器来存储最小的k个数字。
2.每次从输入的n个整数中读取一个数,此时如果容器中已有的数字小于k个,则直接将 读入的数字放入容器,
否则,在容器中找出已有的k个数中的最大值,将读取的数字和最大值比较,如果待插入的值比当前已有的最大值小,则用这个数字替换当前的最大值。
如果待插入的值大于容器中的最大值,则舍弃这个数字。
3.此处可以采用红黑树来实现容器,STL中set和multiset都是基于红黑树实现的,已经对数据自动排序,而且multiset支持重复的数字,所以这里我们使用multiset。
*/

#include<iostream>#include<set>#include<vector>using namespace std;typedef multiset<int,greater<int> >  intset;typedef multiset<int,greater<int> >::iterator setIterator;void GetLeastNumbers(const vector<int>&data,intset& leastNumbers,int k){    leastNumbers.clear();    if(k<1||data.size()<k)        return ;    vector<int>::const_iterator iter = data.begin();    for(;iter!=data.end();++iter)    {        if(leastNumbers.size()<k)            leastNumbers.insert(*iter);        else        {             setIterator  iterGreatest = leastNumbers.begin();             if(*iter<*(leastNumbers.begin()))             {                 leastNumbers.erase(iterGreatest);                 leastNumbers.insert(*iter);             }        }    }}int main(){    int arr[] = {4,5,1,6,2,7,3,8,13,10};    vector<int>it(begin(arr),end(arr));    multiset<int,greater<int> > inset;    GetLeastNumbers(it,inset,6);    for(multiset<int,greater<int> >::iterator data = inset.begin(); data != inset.end(); data++ )    //用迭代器的方式输出容器对象的值    {        cout<<*data<<" ";    }    cout<<endl;}#endif
原创粉丝点击