nth_element 找第k大的数

nth_element

nth_element用于排序一个区间，它使得位置n上的元素正好谁全排序情况下的第n个元素，而且，当nth_element返回的时候，所有按照全排序规则排在位置n之前的元素也都排在位置n之前，按照全排序规则排在n之后的元素全都排在位置n之后。
所以，我们使用nth_element既可以寻找最好的前k个元素，也可以寻找第k大元素。

先看示例：

#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
int main()
{
    int a[10] = { 8, 9, 1, 2, 4, 3, 5, 6, 7, 10 };
    nth_element(a, a + 5, a + 10, std::greater<int>());
    cout << "数组中的中间元素是" << a[5] << endl;
    nth_element(a, a, a + 10, std::greater<int>());
    cout << "数组中第1大元素为" << a[0] << endl;
    nth_element(a, a + 1, a + 10, std::greater<int>());
    cout << "数组中第2大元素是" << a[1] << endl;
}

结果为：

数组中的中间元素是5
数组中第1大元素为10
数组中第2大元素是9
请按任意键继续. . .