C++实现快速选择算法(Quick Select) _ 寻找第K大元素【源码】

老伟大大_4085

老伟大大_4085

2026-04-12

331人浏览

原创

quickselect 比 std::sort 更适合找第k大元素,因其平均时间复杂度为 o(n),不排序只保证第k位就位;而 std::sort 为 o(n log n)。

c++实现快速选择算法(quick select) _ 寻找第k大元素【源码】

为什么 quickselectstd::sort 更适合找第K大元素

因为排序整个数组是 O(n log n),而 quickselect 平均只要 O(n) —— 它不排序,只保证第K位“就位”。实际中,当 K 接近 1 或 n(比如找最大、最小、前3大),quickselect 常比堆或排序快得多,尤其数据量大且不要求稳定时。

但注意:最坏情况是 O(n²),发生在每次选的 pivot 都是最小/最大值。所以必须随机化 pivot,否则退化成冒泡级性能。

怎么写一个健壮的 quickselect(C++ 版本)

核心是复用 partition 逻辑,但只递归处理含目标索引的那一侧。C++ 中推荐用迭代写法避免栈溢出,或至少加尾递归优化。

C++ 算法竞赛自动化测试数据生成与校验框架
C++ 算法竞赛自动化测试数据生成与校验框架

根据原题生成新题面、验证器及完整测试数据,自动套用 testlib 模板,用于用户要求生成测试数据时。

下载
  • 输入数组建议传引用,避免拷贝;若不能修改原数组,先 std::vector 拷贝一份
  • 第K大 → 转为找“升序下标为 n - k”的元素(0-indexed),别硬写降序比较
  • partition 用 Lomuto 方式更易懂,但 Hoare 更省交换次数;实践中 Lomuto + 随机 pivot 足够稳
  • 务必在 partition 前 swap 一次随机位置到末尾,否则 std::vector 的有序输入会触发最坏情况
int quickselect(std::vector<int>& nums, int left, int right, int k) {
    while (left  k) right = mid - 1;
        else left = mid + 1;
    }
    return nums[left];
}</int>

partition 函数里最容易错的三个细节

不是所有 partition 实现都等价。C++ 中若用 判定,返回的 pivot 位置必须满足:左边 ≤ pivot,右边 ≥ pivot,否则二分逻辑会漏掉边界元素。

  • 循环变量用 ij 时,别混淆 “已处理区间” 和 “待扫描区间” 的闭开关系
  • swap 后 i 必须自增,否则重复比较同一元素(常见 off-by-one)
  • 最后 swap nums[i]nums[right] 时,要确认 i 是第一个 ≥ pivot 的位置 —— 这决定了返回值是否能准确划分区间
int partition(std::vector<int>& nums, int left, int right) {
    int pivot = nums[right];
    int i = left;
    for (int j = left; j <h3>调用时绕不开的边界和类型陷阱</h3>
<p><code>k</code> 是“第K大”,但数组索引从 0 开始,且 <code>vector.size()</code> 返回 <code>size_t</code> —— 混用 signed/unsigned 会导致静默翻转(比如 <code>k=1</code> 时 <code>n-k</code> 变成极大正数)。</p>
<ul>
<li>统一用 <code>int</code> 接收 <code>k</code>,计算目标下标前强转:<code>int target = static_cast<int>(nums.size()) - k;</int></code>
</li>
<li>检查 <code>k</code> 是否越界:<code>if (k  nums.size()) throw std::out_of_range("k out of range");</code>
</li>
<li>单元素数组或空数组必须提前处理,否则 <code>rand() % 0</code> 是未定义行为</li>
<li>如果编译器没开 <code>-stdlib=libc++</code> 或 Windows 下,<code>srand(time(0))</code> 只需调一次,别塞进函数里反复调</li>
</ul>
<p>快速选择真正难的不是算法本身,而是 pivot 随机化是否生效、边界是否全覆盖、以及 signed/unsigned 类型混用带来的隐性崩溃 —— 这些地方一错,结果可能偶尔对、偶尔错,极难复现。</p></int>

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

相关专题

更多
sort排序函数用法
sort排序函数用法

sort排序函数的用法:1、对列表进行排序,默认情况下,sort函数按升序排序,因此最终输出的结果是按从小到大的顺序排列的;2、对元组进行排序,默认情况下,sort函数按元素的大小进行排序,因此最终输出的结果是按从小到大的顺序排列的;3、对字典进行排序,由于字典是无序的,因此排序后的结果仍然是原来的字典,使用一个lambda表达式作为key参数的值,用于指定排序的依据。

2023.09.04

1098

7

string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.02

5099

3

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.08.29

2605

6

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

2025.08.29

3168

10

C++中int的含义
C++中int的含义

本专题整合了C++中int相关内容,阅读专题下面的文章了解更多详细内容。

2025.08.29

2225

10

堆和栈的区别
堆和栈的区别

堆和栈的区别:1、内存分配方式不同;2、大小不同;3、数据访问方式不同;4、数据的生命周期。本专题为大家提供堆和栈的区别的相关的文章、下载、课程内容,供大家免费下载体验。

2023.07.18

4507

5

堆和栈区别
堆和栈区别

堆(Heap)和栈(Stack)是计算机中两种常见的内存分配机制。它们在内存管理的方式、分配方式以及使用场景上有很大的区别。本文将详细介绍堆和栈的特点、区别以及各自的使用场景。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

2023.08.10

2068

6

堆和栈的区别
堆和栈的区别

堆和栈的区别:1、内存分配方式不同;2、大小不同;3、数据访问方式不同;4、数据的生命周期。本专题为大家提供堆和栈的区别的相关的文章、下载、课程内容,供大家免费下载体验。

2023.07.18

4507

5

堆和栈区别
堆和栈区别

堆(Heap)和栈(Stack)是计算机中两种常见的内存分配机制。它们在内存管理的方式、分配方式以及使用场景上有很大的区别。本文将详细介绍堆和栈的特点、区别以及各自的使用场景。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

2023.08.10

2068

6

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Conan 2 Essentials 免费课程
Conan 2 Essentials 免费课程

共0课时 | 0人学习

CMake 与 Conan 集成实践
CMake 与 Conan 集成实践

共0课时 | 0人学习

Conan 2 高级依赖模型介绍
Conan 2 高级依赖模型介绍

共0课时 | 0人学习