最小的K个数

输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,

Tips:

使用set模板函数时注意:set不能用下标,只能用迭代器来进行操作,insert和erase

1.:利用堆排序,O(N logK),适合处理海量数据
(1) 遍历输入数组,将前k个数插入到推中;(利用multiset来做为堆的实现)
(2) 继续从输入数组中读入元素做为待插入整数,并将它与堆中最大值比较:如果待插入的值比当前已有的最大值小,则用这个数替换当前已有的最大值;如果待插入的值比当前已有的最大值还大,则抛弃这个数,继续读下一个数。
这样动态维护堆中这k个数,以保证它只储存输入数组中的前k个最小的数,最后输出堆即可。
class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        set<int>s;
         set<int>::iterator it;
        vector<int>a;
        if(input.empty()||k<=0)
            return a;
        
        for(int i=0;i<input.size();i++)
        {
            s.insert(input[i]);
        }
        if(k<=s.size())
        {
         for( it=s.begin();it!=s.end();it++)
         {if(a.size()<k)
         { a.push_back(*it); }
         }
        }
        
        return a;
    }
};

2.我的方法,简单粗暴class Solution {public:    vector<int> GetLeastNumbers_Solution(vector<int> input, int k)    {        vector<int> result;        int len = input.size();        if(input.empty() || k<=0 || len < k) return result;                 multiset<int, greater<int> > leastNumbers; // 从大到小排序        multiset<int, greater<int> >::iterator iterGreater; // 定义迭代器                 vector<int>::iterator iter = input.begin();        for(; iter != input.end(); ++iter)        {            // 将前k个数直接插入进multiset中,注意是小于K            if(leastNumbers.size() < k)            {                leastNumbers.insert(*iter);            }            else            {                // 因为设置的从大到小排序,故multiset中第一个位置的元素即为最大值                iterGreater = leastNumbers.begin();                                 // 如果input中当前元素比multiset中最大元素小,则替换;即保持multiset中这k个元素是最小的。                if(*iter < *(leastNumbers.begin()))                {                    // 替换掉当前最大值                    leastNumbers.erase(iterGreater);                    leastNumbers.insert(*iter);                }            }        }                 for(iterGreater = leastNumbers.begin();iterGreater!=leastNumbers.end();++iterGreater)        {            result.push_back(*iterGreater); // 将multiset中这k个元素输出        }                 return result;    }};


3.
利用快速排序中的获取分割(中轴)点位置函数getPartitiion。
基于数组的第k个数字来调整,使得比第k个数字小的所有数字都位于数组的左边,比第k个数字大的所有数字都位于数组的右边。调整之后,位于数组左边的k个数字就是最小的k个数字(这k个数字不一定是排序的)。O(N)


class Solution {
public:
    void swap(int &fir,int &sec)
    {
        int temp = fir;
        fir = sec;
        sec = temp;
    }
     
    int getPartition(vector<int> &input,int start,int end)
    {
        if(input.empty() || start>end) return -1;
        int temp = input[end];
        int j = start - 1;
        for(int i=start;i<end;++i)
        {
            if(input[i]<=temp)
            {
                ++j;
                if(i!=j) swap(input[i],input[j]);                   
            }
        }
        swap(input[j+1],input[end]);
        return (j+1);
    }
         
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k)
    {
        vector<int> result;       
        if(input.empty() || k>input.size() || k<=0) return result;
         
        int start = 0;
        int end = input.size()-1;
        int index = getPartition(input,start,end);
         
        while(index != (k-1))
        {
            if(index > (k-1))
            {
                end = index - 1;
                index = getPartition(input,start,end);
            }
            else
            {
                start = index + 1;
                index = getPartition(input,start,end);
            }
        }
         
        for(int i=0;i<k;++i)
        {
            result.push_back(input[i]);
        }
         
        return result;
    }
};






  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值