剑指 Offer 40. 最小的k个数 C++ 2021-08-11

最新推荐文章于 2022-08-03 19:22:37 发布

原创最新推荐文章于 2022-08-03 19:22:37 发布 · 135 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

42 篇文章

订阅专栏

本文探讨了三种不同的解决最小k个数问题的方法：使用直接排序，结合快速排序的递归策略，以及利用优先队列的数据结构。介绍了每种方法的原理和代码实现，并展示了如何通过优先队列实现高效的选择操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

剑指 Offer 40. 最小的k个数
在这里插入图片描述

1、直接法排序选k个

class Solution {
public:    
	 vector<int> getLeastNumbers(vector<int>& arr, int k) 
     {
     	 vector<int> res;
       	 sort(arr.begin(), arr.right());
       	 res.assign(arr.begin(), arr.begin() + k);
         return res;
     }
};

2、借助快排+递归

class Solution 
{
public:
    int partition(vector<int>& arr,int left, int right);
    void randomized_selected(vector<int>& arr, int left, int right, int k);
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k);
};

int Solution::partition(vector<int>& arr, int left, int right)
{
    int temp = arr[left];
    while (left < right)
    {
        while (left < right && arr[right] >= temp) 
        {
            right--;
        }
        arr[left] = arr[right];

        while (left < right && arr[left] <= temp) 
        {
            left++;
        }
        arr[right] = arr[left];
    }
    arr[left] = temp;
    return left;
}

void Solution::randomized_selected(vector<int>& arr, int left, int right, int k) 
{
if (left >= right) 
    {
        return;
    }
    int pos = partition(arr, left, right);
    int num = pos - left + 1;
    if (k == num) 
    {
        return;
    } 
    else if (k < num) 
    {
        randomized_selected(arr, left, pos - 1, k);
    } 
    else 
    {
        randomized_selected(arr, pos + 1, right, k - num);
    }
}


vector<int> Solution::getLeastNumbers(vector<int>& arr, int k)
{
    vector<int> res;
    randomized_selected(arr, 0, arr.size() - 1, k);
    res.assign(arr.begin(), arr.begin() + k);
    return res;
}

3、优先队列，选择大顶堆，优先队列还是个队列只能弹出队头元素

class Solution 
{
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k)
    {
        vector<int> res;
        if (k == 0)//注意边界条件 当然也可以加上 if (k > arr.size())的判断
            return res;

        priority_queue<int, vector<int>> priq;
        for (int i = 0; i < k; ++i)
        {
            priq.push(arr[i]);
        }
        int i = k;//注意这个i = k 不要写成i = 0;
        while (i < arr.size())
        {
            if (arr[i] < priq.top())
            {
                priq.pop();
                priq.push(arr[i]);
            }
            i++;
        }
        while (!priq.empty())
        {
            res.push_back(priq.top());
            priq.pop();
        }
        return res;

    }
};

优先队列的补充：关于优先队列的格式