神奇的STL —— nth_element (查询序列第k小)

在今天的模拟赛中,有一道题需要在(mathcal{O}(n))的复杂度内求出一个无序的序列的前(k)大值的和

然后我冥思苦想良久,还是想不到怎么做,最后只能(sort)水分了...

其实C++的模板库(algorithm)里,提供了一个非常好用的函数叫(nth\_element)

调用方式为(nth\_element(a + l, a + k, a + r + 1)),意思就是在(a)数组中的区间([l, r])之间第(k)小的值放到第(k)个位置,比这个第(k)小值小的放到它的前面。同理,比这个第(k)小值大的放到它的后面

然后就可以愉快的调用了

至于原理,可以去看看这篇解释 传送门

原文地址:https://www.cnblogs.com/xunzhen/p/9748269.html