C++STL库:nth_element()
头文件
C++11
#include <algorithm>
语法格式
//排序规则采用默认的升序排序
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:用于自定义排序规则。
细节
- 当采用默认的升序排序规则时,该函数可以从某个序列中找到第 k 小的元素 e(序列下标从0开始),并将 e 移动到序列中第 k 的位置处。不仅如此,整个序列经过 nth_element() 函数处理后,所有位于 e 之前的元素都比 e 小,所有位于 e 之后的元素都比 e 大。
具体实例
作用:查找第k小(大)的元素。
#include<iostream>
#include<algorithm>
using namespace std;
bool mycmp(int x, int y) {
return x > y;
}
int main() {
int a[9] = {0, 1, 2, 3, 4, 5, 6, 7, 8};
int b[9] = {2, 3, 0, 5, 7, 1, 8, 4, 6};
int k = 2;
nth_element(b, b + k, b + 9);
cout<<"第"<<k<<"小的数是:"<<b[k]<<endl;
for(int i = 0; i < 9; i++) {
cout<<b[i]<<" ";
}
cout<<endl;
k = 5;
nth_element(b, b + k, b + 9);
cout<<"第"<<k<<"小的数是:"<<b[k]<<endl;
for(int i = 0; i < 9; i++) {
cout<<b[i]<<" ";
}
cout<<endl;
k = 3;
vector<int> myvector{3, 1, 0, 2, 6, 4, 5, 7, 8};
nth_element(myvector.begin(), myvector.begin() + k, myvector.end(), mycmp);
cout<<"第"<<k<<"大的数是:"<<myvector[k]<<endl;
for(int i = 0; i < 9; i++) {
cout<<myvector[i]<<" ";
}
cout<<endl;
return 0;
}
输出:
第2小的数是:2
0 1 2 3 4 5 6 8 7
第5小的数是:5
4 1 2 3 0 5 6 7 8
第3大的数是:5
6 8 7 5 4 3 2 0 1