C++ nth_element()用法详解，STL排序小技巧

更多资源请关注纽扣编程微信公众号

在有序序列中，我们可以称第 n 个元素为整个序列中“第 n 大”的元素。比如，下面是一个升序序列：
2 4 6 8 10

在这个序列中，我们可以称元素 6 为整个序列中“第 3 小”的元素，并位于第 3 的位置处；同样，元素 8 为整个序列中“第 4 小”的元素，并位于第 4 的位置处。

简单的理解 nth_element() 函数的功能，当采用默认的升序排序规则（std::less）时，该函数可以从某个序列中找到第 n 小的元素 K，并将 K 移动到序列中第 n 的位置处。不仅如此，整个序列经过 nth_element() 函数处理后，所有位于 K 之前的元素都比 K 小，所有位于 K 之后的元素都比 K 大。

当然，我们也可以将 nth_element() 函数的排序规则自定义为降序排序，此时该函数会找到第 n 大的元素 K 并将其移动到第 n 的位置处，同时所有位于 K 之前的元素都比 K 大，所有位于 K 之后的元素都比 K 小。

以下面这个序列为例：
3 4 1 2 5

假设按照升序排序，并通过 nth_element() 函数查找此序列中第 3 小的元素，则最终得到的序列可能为：
2 1 3 4 5

显然，nth_element() 函数找到了第 3 小的元素 3 并将其位于第 3 的位置，同时元素 3 之前的所有元素都比该元素小，元素 3 之后的所有元素都比该元素大。

nth_element() 函数有以下 2 种语法格式：
//排序规则采用默认的升序排序
void nth_element (RandomAccessIterator first,
RandomAccessIterator nth,
RandomAccessIterator last);
//排序规则为自定义的 comp 排序规则
void nth_element (RandomAccessIterator first,
RandomAccessIterator nth,
RandomAccessIterator last,
Compare comp);
其中，各个参数的含义如下：
first 和 last：都是随机访问迭代器，[first, last) 用于指定该函数的作用范围（即要处理哪些数据）；
nth：也是随机访问迭代器，其功能是令函数查找“第 nth 大”的元素，并将其移动到 nth 指向的位置；
comp：用于自定义排序规则。

第二小的数

#include<bits/stdc++.h>

using namespace std;
int main(){
int a[5]={3,4,1,2,5};
nth_element(a,a+1,a+5);//从0开始 第2小
for(int i=0;i<5;i++){
cout<<a[i]<<" ";
}
cout<<endl;
cout<<a[1];
return 0;
}

/*
1 2 3 4 5
2
*/

第二大的数

#include<bits/stdc++.h>

using namespace std;

bool cmp(int n1,int n2){
return n1>n2;
}
int main(){
int a[5]={3,4,1,2,5};
nth_element(a,a+1,a+5,cmp);//从0开始 第2大
for(int i=0;i<5;i++){
cout<<a[i]<<" ";
}
cout<<endl;
cout<<a[1];
return 0;
}

/*
5 4 1 2 3
4
*/