求最小的K个数的三种方法

最小的K个数
给定一个长度为 n 的可能有重复值的数组,找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4(任意顺序皆可)。

1、最简单
全排序。利用各种排序方法,比如冒泡,快速排序,归并排序、希尔排序等。(关于几种排序方法可以仔细学一遍)。复杂度要么n平方,最好也是nlogn。代码就不写了。

2、堆排序
建立一个容量为k的大根堆的优先队列。遍历一遍元素,如果队列大小<k,就直接入队,否则,让当前元素与队顶元素相比,如果队顶元素大,则出队,将当前元素入队。
大根堆用于升序排序(所以求最小的前k个数用大根堆)。堆排序的时间复杂度是O(Nlogk),空间复杂度是O(1),所以在海量数据及内存不足的条件下,该排序方法适用。
优先队列的底层是用堆来实现的。大根堆就是父节点比子节点大的完全二叉树。C++中STL中的优先队列默认的是大根堆,如果要用小根堆则需要加入greater.

priority_queue<Type, Container, Functional>Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector,因为玩完全二叉树用数组比较好表示),Functional 就是比较的方式,当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆。如果要用小根堆则需要加入greater

//升序队列
priority_queue <int,vector,greater > q; 小根堆
//降序队列
priority_queue <int,vector,less >q; 大根堆
所以求最小的K个数要用降序队列,即根是最大的,之后每个数和根比较,如果更大,直接略过,比根小,把根删去,把这个数加进去。

lass Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> ans;
        if(k==0||k>input.size())
            return ans;
        priority_queue<int,vector<int>,less<int>> pq;
        for(int i=0;i<input.size();i++)
        {
            if(pq.size()<k)
                pq.push(input[i]);
            else
            {
                if(input[i]>pq.top())
                    continue;
                else
                {
                    pq.pop();
                    pq.push(input[i]);
                }
            }
        }
        while(!pq.empty())
        {
            ans.push_back(pq.top());
            pq.pop();
        }
        return ans;
        
    }
};

第三种利用快排,可能思考有难度,就不推荐了,但是时间复杂度是O(N)。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值