C++ STL 数据结构与算法 —— 排序

xiaoxiao2021-02-28  45

1. Top k 大的数

排序后直接索引输出: O(nlogn) O ( n log ⁡ n )

std::sort(v.rbegin(), r.rend()); return v[k-1];

使用优先队列( O(klogn) O ( k log ⁡ n ) ,每 pop 一个元素,都需重排大顶堆):

std::priority_queue<int> PQ(v.rbegin(), v.rend()); for (size_t i = 0; i < k-1; ++i) { PQ.pop(); } return PQ.top();

使用自带函数(随机选择算法,期望耗时 O(n) O ( n ) ,最坏 O(n2) O ( n 2 ) ):

std::nth_element(v.rbegin(), v.rbegin()+k-1, v.rend()); return v[k-1];
转载请注明原文地址: https://www.6miu.com/read-2624354.html

最新回复(0)