剑指offer. 40 最小的k个数 top(k)(重要)

剑指offer. 40 最小的k个数 top(k)(重要)

题目描述:

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

解题思路:

解法1: 快速选择,定位下标为k-1的数,O(n)

解法2: 海量数据top k问题。 大(小)顶堆(O(nlogk))

假如求最小的k个,用大顶堆  priority_queue<int> big_heap;

假如求最大的k个,用小顶堆 priority_queue<int,vector<int>,greater<int> > small_heap。

比如这一题最小的k个,那用大顶堆,来了一个新的假如比堆顶heap.top(),也就是堆里面最大的还要小,那堆顶滚出去,这个加进来。否则略过 ,下一个。

解法1代码:

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len=input.size();
        
        if(k<=0 || len==0 || len<k)
            return {};
        int left=0,right=len-1;
        int index=partition(input,left,right);
        while(index !=k-1){
            if(k-1<index){
                right=index-1;
            } else{
                laeft=index+1;
            }
            index=partition(input,left,right);
        }
        vector<int> res(k);
        for(int i=0;i<k;++i){
            res[i]=input[i];
        }
        return res;
        
    }
    
    int partition(vector<int> & vec, int left,int right){
        int pivot=vec[left];
        int less=left;
        for(int i=left+1;i<=right;++i){
            if(vec[i]<pivot)
                swap(vec[i],vec[++less]);
        }
        
        swap(vec[left],vec[less]);
        return less;
    }
    
};

解法2代码:

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len =input.size();
        if(k<=0 || len==0 || len< k )
            return {};      
        vector<int> res;
        priority_queue<int> helper;
        for(int i=0;i!=k;++i){
            helper.push(input[i]);
        }
        for(int i=k;i!=len;++i){
            if(input[i]<helper.top()){
                helper.pop();
                helper.push(input[i]);
            }
        }
        for(int i=0;i!=k;++i){
            res.push_back(helper.top());
            helper.pop();
        }
        return res;
         
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值