讲师中心 微信公众号
AI工具推荐 视频效率加速

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

风杰酱_1106

风杰酱_1106

发布时间:2026-04-11 20:12:15

|

860人浏览过

|

来源于php中文网

原创

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()(低质量、非线程安全)。

立即学习“C++免费学习笔记(深入)”;

C++14
C++14

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

下载
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 < pivot_idx) return quickselect(arr, left, pivot_idx - 1, k);
    else return quickselect(arr, pivot_idx + 1, right, k);
}

std::nth_element 的边界行为和 const 安全性

它不保证“前 k 个元素有序”,只保证第 k 个位置放对了值,且左边 ≤ 它、右边 ≥ 它——这对中位数足够,但别指望它返回一个排好序的半数组。

它会修改原容器,所以传入 const std::vector<int>& 编译失败;若原始数据不可改,必须先拷贝一份(std::vector<int> copy = original;),这点常被忽略导致运行时崩溃或静默错误。

  • 当 k 超出 [0, size) 范围,行为未定义(不是抛异常,是 UB)——务必检查 k = size/2 是否合法(空容器时 size()==0)
  • 对于 float 或自定义类型,需确保 < 可用且满足 strict weak ordering;用 std::nth_element(..., std::greater<int>{}) 可反向找第 k 大
  • 在性能敏感场景(如高频调用),避免反复构造 vector 拷贝——可复用 buffer 或用 std::span(C++20)约束视图

中位数计算中容易被绕晕的索引细节

中位数位置取决于“第几小”而非“下标是否居中”。例如 {1,2,3,4}(size=4),数学中位数是 (2+3)/2=2.5,对应第 2 小(index=1)和第 3 小(index=2);而 size/2 是 2(0-indexed 下标),刚好是后者。但若误用 (size-1)/2 在奇数时又会错位。

统一做法:定义 k1 = (size-1)/2,k2 = size/2(整数除法),两者在奇数时相等,偶数时相邻——这样不用分支判断奇偶。

  • 奇数长度:std::nth_element(v.begin(), v.begin() + v.size()/2, v.end()) 后取 v[v.size()/2]
  • 偶数长度:必须分别获取 v[(size-1)/2] 和 v[size/2],不能只调一次 nth_element 然后读两个位置——因为 partition 后只有目标位置确定,邻位不保序
  • 用 std::vector::data() + std::nth_element 可避免迭代器开销,尤其对大数组

实际项目里,95% 的中位数需求用 std::nth_element 加两行索引逻辑就够了;手写快速选择只在需要完全控制 pivot 策略、或嵌入式环境无 STL 时才值得投入。真正难的是把索引算对,而不是算法本身。

热门AI工具

更多
讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

相关专题

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

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

6487

2024.04.28

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

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

744

2025.10.23

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

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

1138

2023.09.04

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

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

2198

2023.09.20

string转int
string转int

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

6299

2023.08.02

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

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

3065

2024.08.29

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

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

3948

2025.08.29

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

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

2745

2025.08.29

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

40

2026.10.16

热门下载

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

精品课程

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

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

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

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn