【C++】n_element的用法

n_element函数定义在头文件#include <algorithm>里面，用法是：

n_element(v.begin(), v.begin() + k, v.end());

表示假设v排序后v[k]应该存储的值为v[k]，则n_element函数的作用就是让v[k]这个值放在v[k]处，且让v[k]左边的数都小于v[k]，v[k]右边的数都大于v[k]。但是不保证他们是有序的。（即v[k]处存放的是第k+1大的数，并且左边数都比他小右边数字都比他大，比如假设k = 1，表示将v[1]处设置为第2大的数）

也可以自定义添加cmp函数，表示方法为：

n_element(v.begin(), v.begin() + k, v.end(), cmp);

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
    vector<int> v{5, 6, 4, 3, 2, 6, 7, 9, 3};
    nth_element(v.begin(), v.begin() + v.size()/2, v.end());
    cout << "The median is " << v[v.size()/2] << '\n';
}

#include <iostream>

#include <vector>

#include <algorithm>

using namespace std;

int main() {

vector<int> v{5, 6, 4, 3, 2, 6, 7, 9, 3};

nth_element(v.begin(), v.begin() + v.size()/2, v.end());

cout << "The median is " << v[v.size()/2] << '\n';

}

输出为：

The median is 5

❤ 点击这里 -> 订阅《PAT | 蓝桥 | LeetCode学习路径 & 刷题经验》by 柳婼

❤ 点击这里 -> 订阅《从放弃C语言到使用C++刷算法的简明教程》by 柳婼

❤ 点击这里 -> 订阅PAT甲级乙级、蓝桥杯、GPLT天梯赛、LeetCode题解离线版