C++ nth_element()用法详解

在有序序列中,我们可以称第 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
*/
posted @ 2022-11-09 10:51  new-code  阅读(166)  评论(0编辑  收藏  举报