C++STL库：nth_element()

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