C++实现快速选择算法查找中位数 _ 均摊O(n)复杂度实现【源码】

风杰酱_1106

风杰酱_1106

2026-04-11

860人浏览

原创

std::nth_element是找中位数的首选,因其均摊o(n)、内置优化且规避最坏情况;需注意索引计算、容器可修改性及随机访问要求。

c++实现快速选择算法查找中位数 _ 均摊o(n)复杂度实现【源码】

为什么 std::nth_element 是找中位数的首选

直接调用 std::nth_element 就能均摊 O(n) 找到中位数,不用手写 partition 循环或递归——它内部就是快速选择(Quickselect)的工业级实现,做了三数取中、小数组切片优化、尾递归消除,还规避了最坏 O(n²) 的常见坑。

常见错误是误用 std::sort 或 std::partial_sort:前者 O(n log n) 过重,后者在找单个中位数时仍做多余排序;还有人手动实现但忘了随机打乱,一遇到已排序数组就退化。

  • 对偶数长度数组,中位数定义为第 n/2 小(0-indexed),即 std::nth_element(v.begin(), v.begin() + v.size()/2, v.end())
  • 若需严格按数学定义(偶数时取中间两数平均),先取第 (n-1)/2 和 n/2 小两个位置,再平均——注意两次调用 std::nth_element 会改变原数组顺序,应只调一次 + 一次扫描比较
  • 输入容器必须支持随机访问迭代器(std::vector、std::array 可,std::list 不可)

手写快速选择时必须加随机化 pivot

不随机选 pivot 的快速选择,在已排序或逆序数组上必退化成 O(n²)。C++ 标准库实现默认随机,但自己写时容易忽略这一步。

实操建议:用 std::random_device 和 std::mt19937 生成索引,交换到末尾再 partition;不要用 rand()(低质量、非线程安全)。

int quickselect(std::vector<int>& arr, int left, int right, int k) {
    if (left == right) return arr[left];
    std::random_device rd;
    std::mt19937 g(rd());
    std::uniform_int_distribution<int> dist(left, right);
    std::swap(arr[dist(g)], arr[right]); // 随机换 pivot 到末尾
    int pivot_idx = partition(arr, left, right);
    if (k == pivot_idx) return arr[k];
    else if (k <h3>
<code>std::nth_element</code> 的边界行为和 const 安全性</h3>
<p>它不保证“前 k 个元素有序”,只保证第 k 个位置放对了值,且左边 ≤ 它、右边 ≥ 它——这对中位数足够,但别指望它返回一个排好序的半数组。</p><div class="aritcle_card flexRow artxards">
											<div class="artcardd flexRow">
												<a class="aritcle_card_img" rel="nofollow" href="/xiazai/gongju/2823" title="C++14"><img
														src="https://img.php.cn/upload/manual/001/431/639/6ac8b33c327c4749.png" alt="C++14" onerror="this.onerror='';this.src='/static/lhimages/moren/morentu.png'" ></a>
												<div class="aritcle_card_info flexColumn">
													<a rel="nofollow" href="/xiazai/gongju/2823" title="C++14" class="overflowclass">C++14</a>
													<p class="overflowclass">C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。</p>
												</div>
												<a rel="nofollow" href="/xiazai/gongju/2823" title="C++14" class="aritcle_card_btn flexRow flexcenter"><b></b><span>下载</span>
												</a>
											</div>
										</div>
<p>它会修改原容器,所以传入 <code>const std::vector<int>&</int></code> 编译失败;若原始数据不可改,必须先拷贝一份(<code>std::vector<int> copy = original;</int></code>),这点常被忽略导致运行时崩溃或静默错误。</p>
<ul>
<li>当 <code>k</code> 超出 <code>[0, size)</code> 范围,行为未定义(不是抛异常,是 UB)——务必检查 <code>k = size/2</code> 是否合法(空容器时 <code>size()==0</code>)</li>
<li>对于 <code>float</code> 或自定义类型,需确保 <code> 可用且满足 strict weak ordering;用 <code>std::nth_element(..., std::greater<int>{})</int></code> 可反向找第 k 大</code>
</li>
<li>在性能敏感场景(如高频调用),避免反复构造 vector 拷贝——可复用 buffer 或用 <code>std::span</code>(C++20)约束视图</li>
</ul>
<h3>中位数计算中容易被绕晕的索引细节</h3>
<p>中位数位置取决于“第几小”而非“下标是否居中”。例如 <code>{1,2,3,4}</code>(size=4),数学中位数是 <code>(2+3)/2=2.5</code>,对应第 2 小(index=1)和第 3 小(index=2);而 <code>size/2</code> 是 2(0-indexed 下标),刚好是后者。但若误用 <code>(size-1)/2</code> 在奇数时又会错位。</p>
<p>统一做法:定义 <code>k1 = (size-1)/2</code>,<code>k2 = size/2</code>(整数除法),两者在奇数时相等,偶数时相邻——这样不用分支判断奇偶。</p>
<ul>
<li>奇数长度:<code>std::nth_element(v.begin(), v.begin() + v.size()/2, v.end())</code> 后取 <code>v[v.size()/2]</code>
</li>
<li>偶数长度:必须分别获取 <code>v[(size-1)/2]</code> 和 <code>v[size/2]</code>,不能只调一次 <code>nth_element</code> 然后读两个位置——因为 partition 后只有目标位置确定,邻位不保序</li>
<li>用 <code>std::vector::data()</code> + <code>std::nth_element</code> 可避免迭代器开销,尤其对大数组</li>
</ul>
<p>实际项目里,95% 的中位数需求用 <code>std::nth_element</code> 加两行索引逻辑就够了;手写快速选择只在需要完全控制 pivot 策略、或嵌入式环境无 STL 时才值得投入。真正难的是把索引算对,而不是算法本身。</p></int></int>

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

相关文章

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

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

下载

相关标签:

c++

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

相关专题

更多
css中float用法
css中float用法

css中float属性允许元素脱离文档流并沿其父元素边缘排列,用于创建并排列、对齐文本图像、浮动菜单边栏和重叠元素。想了解更多float的相关内容,可以阅读本专题下面的文章。

2024.04.28

6487

8

C++中int、float和double的区别
C++中int、float和double的区别

本专题整合了c++中int和double的区别,阅读专题下面的文章了解更多详细内容。

2025.10.23

744

4

sort排序函数用法
sort排序函数用法

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

2023.09.04

1138

7

c语言const用法
c语言const用法

const是关键字,可以用于声明常量、函数参数中的const修饰符、const修饰函数返回值、const修饰指针。详细介绍:1、声明常量,const关键字可用于声明常量,常量的值在程序运行期间不可修改,常量可以是基本数据类型,如整数、浮点数、字符等,也可是自定义的数据类型;2、函数参数中的const修饰符,const关键字可用于函数的参数中,表示该参数在函数内部不可修改等等。

2023.09.20

2198

7

string转int
string转int

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

2023.08.02

6279

3

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

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

2024.08.29

3045

6

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

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

2025.08.29

3928

10

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

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

2025.08.29

2725

10

线程和进程的区别
线程和进程的区别

线程和进程的区别:线程是进程的一部分,用于实现并发和并行操作,而线程共享进程的资源,通信更方便快捷,切换开销较小。本专题为大家提供线程和进程区别相关的各种文章、以及下载和课程。

2023.08.10

4158

6

热门下载

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

精品课程

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

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

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

共0课时 | 0人学习