nth_element()函数用法

STL中的nth_element()函数用法详解

一、原函数

nth_element(first, nth, last, compare)

求[first, last]这个区间中第n大小的元素，如果参数加入了compare函数，就按compare函数的方式比较。

二、头文件

#include<algorithm>

三、作用

nth_element仅排序第n个元素（从0开始索引），即将位置n（从0开始）的元素放在第n大的位置，处理完之后，默认排在它前面的元素都不比它大，排在它后面的元素都不比它小。

例如：array[first, last)元素区间，排序后，array[nth]就是第n大的元素（从0开始）

但是[first, nth) 和 [nth,last)区间的大小顺序不一定。但是可以确定的是array[nth]一定是整个区间里第n大的元素。

[first,nth)中的元素都是不大于array[nth]的，[nth, last)中的元素都是不小于array[nth]的。

四、代码实现

#include<bits/stdc++.h>
using namespace std;
long long n,k,a[5000010];
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    nth_element(a,a+k,a+n);//使第k小整数就位 
    printf("%d",a[k]);//调用第k小整数
}