nth_element实际上就是快速选择算法的实现,快速选择算法实现原理具体见这里。
它是STL里面的函数,使用时需要包含头文件<algorithm>
void nth_element (Iterator first, Iterator first+nth, Iterator last, Compare comp);
重新排列range [first,last)
中的元素,使第n个位置的元素是按排序顺序在该位置的元素。
其他元素没有任何特定的顺序,只是第n个元素之前的元素都不大于该元素,而第n个元素后面的元素均不小于该元素。
#include <iostream> // std::cout
#include <algorithm> // std::nth_element, std::random_shuffle
#include <vector> // std::vector
#include<iomanip>
using namespace std;
bool myfunction (int i,int j) { return (i>j); }
int main() {
std::vector<int> myvector ;
// set some values:
for (int i = 0; i < 100; i++) myvector.push_back(i);
std::random_shuffle(myvector.begin(), myvector.end());
// using default comparison (operator <):
std::nth_element(myvector.begin(), myvector.begin() + 50, myvector.begin()+100);
// using function as comp
//std::nth_element(myvector.begin(), myvector.begin() +10, myvector.end(), myfunction);
// print out content:
std::cout << "myvector contains:";
for (std::vector<int>::iterator it = myvector.begin(); it != myvector.end(); ++it)
{
if ((it - myvector.begin()) % 10 == 0)std::cout << endl;
std::cout << setw(5) << *it;
}
std::cout << '\n';
return 0;
}
输出:
我这里不知为何,数字少了输出结构和直接排序一样,所以就用了100个数。在第50个位置,即myvector[49]处的元素未第50大的数(即49) ,49之前的数全部小于49,之后的数全部大于49。这种算法也可以用来找第k大的数。