c++的stl库之nth_element函数

nth_element函数

好久没用补题了，今天去牛客补了几道水题，其中一道是要在无序表中快速找到第k小的数，这道题数据太大了，用sort()和partial_sort() 都会超时,垃圾的我一开始就是直接用sort写的，不出意料的超时了，接着又尝试了桶排序，自己又tle了，然后无聊的翻了下群，发现有个人说nth_element函数可以快速做完这道题，顿时感觉孤陋寡闻了，马上百度了下这个函数，交了一波，ac了。事后，还是记下笔记，写一下这个函数的用法，免得下次又忘了。

nth_element()

• 头文件：#include<algorithm>
• 格式：nth_element(begin,nth,end,compare)
• 期望复杂度：O(N)

例子：数组a在第1-n位置有元素，现求第k大的数

nth_element(a+1,a+k,a+n+1,cmp)

注意，代码中的cmp可以删去，如果不使用cmp函数，默认是“<”

原理：

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

通过线性的扫描交换，类似快排的一部分，

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

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

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

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

代码：

#include<bits/stdc++.h>
using namespace std;
int a[5000006];
inline int read(){
    int x = 0, f = 1;
    char ch = getchar();
    while(ch < '0' || ch > '9'){
        if (ch == '-')
            f = -1;
        ch = getchar();
    }
    while(ch >= '0' && ch <= '9'){
        x = (x<<1) + (x<<3) + (ch^48);
        ch = getchar();
    }
    return x * f;
}
int main()
{
    
    int t,n,k;
    t=read();
    while(t--)
    {
        n=read();
        k=read();
        k--;
        for(int i=0;i<n;i++)
            a[i]=read();
        nth_element(a,a+k,a+n);
        cout<<a[k]<<"\n";
    }
}