C++实现快速选择算法查找第K大数 _ 平均时间复杂度O(n)解法【源码】

星萱同学_2624

星萱同学_2624

2026-05-09

859人浏览

原创

std::nth_element不能直接交题,因它默认找第k小而非第k大,且会修改原容器顺序;求第k大须用nums.size()-k索引或greater()比较器,否则逻辑错误。

c++实现快速选择算法查找第k大数 _ 平均时间复杂度o(n)解法【源码】

为什么 std::nth_element 不能直接拿去交题?

很多初学者看到 C++ 标准库有 std::nth_element,以为调用它就能“白嫖” O(n) 找第 K 大数——实际中常踩两个坑:它默认找第 K 小(不是第 K 大),且修改原容器顺序,副作用不可忽视。比如你在力扣上写 nth_element(nums.begin(), nums.begin()+k-1, nums.end()),得到的是第 k 小值;若要第 k 大,得传入 greater<int>()</int> 或手动换索引:用 nums.size() - k 作偏移。

手写快速选择必须处理的三个边界

自己实现 quickSelect 时,最容易崩在分区(partition)环节。常见错误包括:

  • 左右指针越界未检查,比如 while (i 忘了 <code>i 判断,导致访问 <code>arr[-1]
  • 分区后 pivot 最终位置没和 right 交换,而是和 i 或 j 盲交换,结果 pivot 不在正确位置
  • 递归调用时区间开闭不一致,比如左半段写成 [left, i-1],右半段却写成 [i+1, right],但实际分区后 pivot 在 i,那左半段应是 [left, i-1],右半段是 [i+1, right] —— 这个没问题;但若你用 j 作 pivot 位置,就得对应改右边界

推荐统一用 Lomuto 分区法:选 arr[right] 为 pivot,用 i 指向小于等于 pivot 的最右位置,循环结束后 swap arr[i] 和 arr[right],返回 i。这样逻辑干净,不易错。

第 K 大 vs 第 K 小:索引别硬背,现场推

给定数组长度 n,求第 K 大数,等价于求升序排列后下标为 n - k 的元素(0-indexed)。但快速选择本身只保证“第 t 小”的位置正确,所以调用时目标索引是 t = n - k。不要记“第 K 大要改成 k-1”,而要每次写:「我要的数,在排好序后从 0 开始数是第几个?」

示例:

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载
vector<int> nums = {3,2,1,5,6,4}; // n=6
k = 2 → 第 2 大是 5 → 升序后 [1,2,3,4,5,6],下标是 4 → t = 6-2 = 4
调用 quickSelect(nums, 0, 5, 4);
</int>

随机化 pivot 是必须项,不是可选项

不随机选 pivot 时,快选退化成 O(n²),比如对已排序数组找最小值,每次 pivot 都是最大值,每次只排除一个元素。C++ 中用 random_device + mt19937 初始化一次即可,不必每次 partition 都重置生成器。

关键代码片段:

int randIdx = left + uniform_int_distribution<int>(0, right - left)(rng);
swap(arr[randIdx], arr[right]);
</int>

注意:rng 应是函数外定义的静态或成员变量 mt19937,否则重复构造开销大,且可能因时间种子相同导致随机性失效。

真正难的不是写对算法,而是想清楚:你要的“第 K 大”在当前无序子数组里,对应的是“第几个小”。这个映射一旦错,后面全错。而且随机化 pivot 如果漏掉,本地小数据跑得飞快,一到大数据就超时,还很难复现。

C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

c++

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
c++和c语言的区别有哪些
c++和c语言的区别有哪些

c++和c语言的区别:1、面向对象编程(OOP)支持不同;2、新增特性不同;3、标准库不同;4、编译方式不同;5、命名空间不同等等。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.14

2228

9

c++和python学习顺序推荐
c++和python学习顺序推荐

一般建议先学习C++,再学习Python,因为这样可以逐步从较为底层的编程语言向更高级的语言过渡。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

999

6

python和c++学习性价比分析
python和c++学习性价比分析

Python易于学习,广泛应用于Web开发、数据科学和人工智能等领域,但性能较低。C语言性能高,适用于对性能要求较高的场景,如游戏开发和系统编程,但学习曲线陡峭,错误处理复杂。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

427

5

c语言和c++一样吗
c语言和c++一样吗

c语言和c++是两种不同的编程语言,虽然有相似之处,但存在显著差异。c语言专注于过程式编程和系统级开发,以简洁、高效著称。c++作为c语言的超集,引入了面向对象编程,增强了代码组织和管理能力,但学习曲线也更陡峭。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

327

5

c语言和c++先学哪个好
c语言和c++先学哪个好

初学者选择学习c语言还是c++语言,需要根据个人学习目标、背景以及编程兴趣和预期应用方向来决定。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

406

5

c语言和c++的区别和联系
c语言和c++的区别和联系

c语言和c++是计算机科学领域应用广泛的编程语言。虽然它们有着相似的基础,但它们在语言类型、语法功能和内存管理方面存在着显著差异。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

580

5

c++软件中文更改教程
c++软件中文更改教程

对于 ide,可通过打开设置,找到语言设置,选择中文,并保存更改。对于非 ide 应用程序,可查找设置或选项,选择语言设置,更改为中文,并保存更改。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.21

1409

9

python和java和c++学习性价比分析
python和java和c++学习性价比分析

Python以其易学性、丰富的库和活跃的社区而著称,适合数据科学、人工智能和Web开发。Java以其跨平台性、企业级应用开发和Android应用开发而闻名。C++以其底层控制能力、高效性能和游戏开发而著称。选择哪种语言取决于个人兴趣、职业方向和特定需求。想了解更多python和java和c++的相关内容,可以阅读本专题下面的文章。

2024.03.22

1197

7

c++和c语言学习顺序推荐
c++和c语言学习顺序推荐

对于初学者,建议先学习C语言,掌握编程基础后再转入C++,便于理解面向对象编程概念。有编程经验者可直接学习C++,快速接触高级编程技术。想了解更多c++和c语言的相关内容,可以阅读本专题下面的文章。

2024.03.25

1325

9

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

C++ 官方标准说明
C++ 官方标准说明

共0课时 | 0人学习