c++备战CCF之力扣简单题笔记（找出最小k个数）

力扣链接：力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台

一、大顶堆

利用c++的STL 优先队列priority_queue，实时记录更新最小的k个数

声明：priority_queue<int> q //c++默认是大顶堆

小顶堆声明：priority_queue<int,vector<int>,greater<int>> q;

代码：

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> vec(k, 0);
        if (k == 0) { // 排除 0 的情况
            return vec;
        }
        priority_queue<int> Q;
        for (int i = 0; i < k; ++i) {
            Q.push(arr[i]);
        }
        for (int i = k; i < (int)arr.size(); ++i) {
            if (Q.top() > arr[i]) {
                Q.pop();
                Q.push(arr[i]);
            }
        }
        for (int i = 0; i < k; ++i) {
            vec[i] = Q.top();
            Q.pop();
        }
        return vec;
    }
};

二、快速选择（快速排序中的前后指针法变种）

快速排序的前后指针法：

1、选出一个key，一般是最左边或是最右边的。
2、起始时，prev指针指向序列开头，cur指针指向prev+1。
3、若cur指向的内容小于key，则prev先向后移动一位，然后交换prev和cur指针指向的内容，然后cur指针++；若cur指向的内容大于key，则cur指针直接++。如此进行下去，直到cur到达end位置，此时将key和++prev指针指向的内容交换即可。

经过一次单趟排序，最终也能使得key左边的数据全部都小于key，key右边的数据全部都大于key。

图像演示：http://t.csdn.cn/84oPO

对于只需要找出最小k个数，只需要随机化基准数，找出基准数是最小的第num个数，若k==num，直接返回；若k<num，则在left~num中间继续寻找；若k>num，则在num~right中间继续寻找。

代码:

class Quick_Select //快速选择
{
    int partition(vector<int>& nums, int l, int r) //以基准值为中心，将数分开
    {
        int pivot = nums[r];//令基准值为随机arr[随机数]
        int i = l - 1;
        for (int j = l; j <= r - 1; ++j) //i记录当前位置，j依次判断是否大于基准值
        {
            if (nums[j] <= pivot)
            {
                i = i + 1;
                swap(nums[i], nums[j]);
            }
        }
        swap(nums[i + 1], nums[r]);//arr[0]--arr[i]为小于基准值的数
        return i + 1;
    }

    // 基于随机的划分
    int randomized_partition(vector<int>& nums, int l, int r)
    {
        int i = rand() % (r - l + 1) + l;//随机 l ~ r-l+1 之间的数
        swap(nums[r], nums[i]); //将基准值放到最后，便于比较
        return partition(nums, l, r);
    }

    void randomized_selected(vector<int>& arr, int l, int r, int k) //基于随机基准找到最小的k个数
    {
        if (l >= r) return;//退出递归
        int pos = randomized_partition(arr, l, r);//随机基准值的位置
        int num = pos - l + 1;//对与新的区间，基准值是第num小的数
        if (k == num)
        {
            return;
        }
        else if (k < num)
        {
            randomized_selected(arr, l, pos - 1, k);
        }
        else
        {
            randomized_selected(arr, pos + 1, r, k - num);//新区间，变成找最小的k-num个数
        }
    }

public:
    vector<int> getLeastNumbers(vector<int>& arr, int k)
    {
        srand((unsigned)time(NULL));//随机种子初始化
        randomized_selected(arr, 0, (int)arr.size() - 1, k);
        vector<int> vec;
        for (int i = 0; i < k; ++i)
        {
            vec.push_back(arr[i]);
        }
        return vec;
    }
};