nth_element

最新推荐文章于 2025-02-01 23:46:46 发布

ac__dreamer

最新推荐文章于 2025-02-01 23:46:46 发布

阅读量1.7k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qq_40093905/article/details/83095820

本文深入解析了C++ STL中的nth_element函数，详细介绍了如何使用该函数进行快速查找数组中的第k个元素，同时对比了nth_element与sort函数在不同场景下的应用。通过具体代码示例，展示了nth_element的平均O(n)复杂度优势。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<map>
#include<unordered_set>
using namespace std;
typedef long long ll;
int a[1000005];
int main()
{

    int i,j,n,m,k;
    long long x,y;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    if(n%2)
    {
        k=(n+1)/2;
        nth_element(a+1,a+k,a+1+n);
        printf("%d\n",a[k]);
    }
    else
    {
        k=n/2;
        nth_element(a+1,a+k,a+1+n);
        x=a[k];
        k++;
        nth_element(a+1,a+k,a+1+n);
        y=a[k];
        x=y+x;
        x>>=1;
        printf("%d\n",(int)x);
    }
    return 0;
}

nth_elemet作为stl类的一个求第k大的数，据说平均复杂度是o(n)；

nth_element(a+1,a+k,a+1+n),从a[1]到a[n]中找第k小的数;

如果找第k大，和sort一样

bool cmp(int a,int b)
{
return a>b;
}

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