C++ nth_element()用法详解
在有序序列中,我们可以称第 n 个元素为整个序列中“第 n 大”的元素。比如,下面是一个升序序列:
2 4 6 8 10
在这个序列中,我们可以称元素 6 为整个序列中“第 3 小”的元素,并位于第 3 的位置处;同样,元素 8 为整个序列中“第 4 小”的元素,并位于第 4 的位置处。
简单的理解 nth_element() 函数的功能,当采用默认的升序排序规则(std::less
当然,我们也可以将 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
*/
作者:newcode 更多资源请关注纽扣编程微信公众号
从事机器人比赛、机器人等级考试、少儿scratch编程、信息学奥赛等研究学习