您的位置:首页 > 编程语言 > C语言/C++

C++ STL nth_element原理与应用

2017-07-09 08:57 288 查看
LZ最近的考试中,某题可以通过调用nth_element()来水过70%的数据

但是LZ并不会啊(历史总是惊人的相似)

于是就有了这篇blog

在编写代码时,有时会有“在一个无序表中快速得到第K小的元素”的需求

而直接排序 不能水过一些测试点 时间不能承受

于是STL的algorithm头文件就给我们提供了nth_element()这样的部分排序函数

调用的正确姿势是这样的:nth_element(first,kth,last)

first和last表示部分排序的区间为[first,last)

指针kth表示完成后,这个位置的元素是第(kth-first+1)小的

(注意first是第1小的)

这个函数的期望复杂度是O(n)的

原理如下:

在当前区间[L,R]上,找一个基准位置mid

通过线性的扫描交换,

使得[L,mid)的元素都比mid小,(mid,R]的元素都比mid大

此时mid上的元素就是第mid小的

然后判断k在哪半边,继续递归处理

所以这样就达到了期望的O(N)复杂度

欢迎补充~~
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: