在做题中学习(79):最小K个数

解法:快速选择算法

**说明:**堆排序也是经典解决问题的算法,但时间复杂度为:O(NlogK),K为k个元素

而将要介绍的快速选择算法的时间复杂度为: O(N)

先看我的前两篇文章,分别学习:数组分三块,随机选择基准值的思想。会的话直接看就完事了

解惑:

1.a,b,c是什么意思?

a,b,c分别是<key, = key, >key 所代表的区间中值的个数

2.如何判断?

看落在哪个区间,a区间全是<key的,所以如果落在这个区间,说明k就在a这个区间,因此就只在这个区间递归即可。

而如果 a + b >=k 说明,k > a了也就是说不仅在a区间,一定也包含b这个区间,而b都是= key的,所以此时直接返回即可,无需继续递归。

如果都不是,说明k > a + b了,所以肯定也落进了c区间,而因为现在我们跳过了 a+b 个元素,所以要找的其实是剩下的k - b - c个元素!继续递归即可。

3.返回值

函数的返回值要求是一个vector,而经过上面的分析,k个元素绝对是在一个区间中的,所以即便递归结束后数组是乱序,只要从[0,k]大小的区间内所有值都符合最小的k个元素,题目也说了可以以任意顺序返回,那结果就是直接返回递归后的[nums.begin(),nums.begin()+k]即可。

附上完整代码:

cpp 复制代码
class Solution 
{
public:
    vector<int> smallestK(vector<int>& nums, int k) 
    {
        srand(time(nullptr));
        qselect(nums,0,nums.size()-1,k);
        return {nums.begin(),nums.begin() + k};
    }

    void qselect(vector<int>& nums,int l,int r,int k)
    {
        if(l >= r)
            return ;
        int key = GetRandomkey(nums,l,r);
        int left = l-1,right = r+1;
        for(int i = l;i<nums.size();)
        {
            if(nums[i] < key)
                swap(nums[++left],nums[i++]);
            else if(nums[i] == key)
                i++;
            else if(nums[i] > key)
            {
                if(i == right)
                    break;
                swap(nums[--right],nums[i]);
            }
        }
        int a = left - l + 1,b = right - left - 1;
        if(a >= k)
            return qselect(nums,l,left,k);
        else if(a + b >=k)
            return;
        else 
            return qselect(nums,right,r,k - a - b);
        
    }

    int GetRandomkey(vector<int>& nums,int l,int r)
    {
        int random = rand();
        return nums[random % (r - l + 1) + l];
    }

};
相关推荐
ゞ 正在缓冲99%…9 分钟前
leetcode76.最小覆盖子串
java·算法·leetcode·字符串·双指针·滑动窗口
xuanjiong10 分钟前
纯个人整理,蓝桥杯使用的算法模板day2(0-1背包问题),手打个人理解注释,超全面,且均已验证成功(附带详细手写“模拟流程图”,全网首个
算法·蓝桥杯·动态规划
Zhichao_9728 分钟前
【UE5 C++课程系列笔记】33——商业化Json读写
c++·ue5
吴梓穆29 分钟前
UE5学习笔记 FPS游戏制作35 使用.csv配置文件
笔记·学习·ue5
惊鸿.Jh29 分钟前
【滑动窗口】3254. 长度为 K 的子数组的能量值 I
数据结构·算法·leetcode
明灯L30 分钟前
《函数基础与内存机制深度剖析:从 return 语句到各类经典编程题详解》
经验分享·python·算法·链表·经典例题
虾球xz32 分钟前
游戏引擎学习第199天
学习·游戏引擎
碳基学AI36 分钟前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义免费下载方法
大数据·人工智能·python·gpt·算法·语言模型·集成学习
补三补四39 分钟前
机器学习-聚类分析算法
人工智能·深度学习·算法·机器学习
独好紫罗兰1 小时前
洛谷题单3-P5718 【深基4.例2】找最小值-python-流程图重构
开发语言·python·算法