菜鸟笔记
提升您的技术认知

剑指offer编程练习029——最小的k个数-ag真人游戏

题目描述

输入n个整数,找出其中最小的k个数。

例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4。

class solution {
public:
    vector getleastnumbers_solution(vector& input, int k){
        int len = input.size();
        vector result;
        if (len <= 0 || k > len || k <= 0)
            return result;
        for (int i = 0; i < k;   i)
            result.push_back(input[i]); //result中最开始存入前k个数
        make_heap (result.begin(),result.end()); //建最大堆
        for (int i = k; i < len;   i){
            if (input[i] < result[0]) { //对后面的数,如果有比堆中最大的数小的就压入堆中
                pop_heap(result.begin(), result.end());
                result.pop_back();
                result.push_back(input[i]);
                push_heap(result.begin(), result.end());
            }
        }
        sort_heap(result.begin(), result.end());
        return result;
    }
};

 

网站地图