剑指offer专题——排序（持续更新）

剑指offer 29——最小的K个数

题目描述

给定一个数组，找出其中最小的K个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。如果K>数组的长度，那么返回一个空的数组

输入
[4,5,1,6,2,7,3,8],4
返回值
[1,2,3,4]

解题思路

使用堆排序（大顶堆、优先队列）
1.当优先队列的size<k时，元素直接入队
2.当等于K的时候，元素与堆顶元素相比，小于堆顶时选择替换堆顶元素
3.循环结束，将堆中元素放入vector，结束

代码

vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
    vector<int> ret;
    if (k==0 || k > input.size()) return ret;
    priority_queue<int, vector<int>> pq;
    //1. 保留最小的K个数
    for (const int val : input) {
        if (pq.size() < k) {
            pq.push(val);
        }
        else {
            if (val < pq.top()) {
                pq.pop();
                pq.push(val);
            }

        }
    }
	//2. 数据写入vector<int>
    while (!pq.empty()) {
        ret.push_back(pq.top());
        pq.pop();
    }
    return ret;
}