partial_sort 比 sort + 取前 n 快,因其时间复杂度为 o(n log k)(k 为取的个数),而后者为 o(n log n);当 k 远小于容器大小时优势显著,但会破坏原容器顺序,且 n 超限时行为未定义。

partial_sort 为什么比 sort + 取前 N 快
因为 partial_sort 内部用的是堆(通常是基于 make_heap + 多次 pop_heap),时间复杂度是 O(N log K)(K 是要取的个数),而先 sort 再取前 N 是 O(N log N)。当 N 远小于容器大小时,性能差距明显——比如从 100 万整数里找前 10 个最大值,partial_sort 实测快 3–5 倍。
注意:它会**破坏原容器顺序**,前 N 个元素有序(升序),其余部分无序,且不保证相对位置。
- 如果你需要“前 N 个最大值”但要求升序排列(即最小的在前、最大的在后),得传入
greater<int>()</int> - 如果你只关心值、不关心原始索引,直接用;若需保留索引,得用
vector<pair size_t>></pair>预处理 - 对
vector、deque等支持随机访问的容器才可用;list不行
获取前 N 个最大值的标准写法
核心是把 partial_sort 的第三个参数设为 begin() + N,并指定降序比较器:
vector<int> v = {5, 2, 9, 1, 7, 3};
size_t N = 3;
partial_sort(v.begin(), v.begin() + N, v.end(), greater<int>());
// v[0..2] 现在是 {9, 7, 5}(降序)
</int></int>
- N 超过容器大小?
partial_sort行为未定义(常见 crash 或乱序),务必提前检查:N = min(N, v.size()) - 比较器必须满足严格弱序;自定义类型记得重载
operator>或传入 lambda:[&](const auto& a, const auto& b) { return a.score > b.score; } - 如果只是读取、不想改原数据,先
copy一份再排,别用const_cast强转
partial_sort_copy 更安全的替代方案
当你不能/不想修改原容器,又不想手动 copy + partial_sort,partial_sort_copy 是更干净的选择:
vector<int> src = {5, 2, 9, 1, 7, 3};
vector<int> dst(3); // 必须预先分配足够空间
partial_sort_copy(src.begin(), src.end(), dst.begin(), dst.end(), greater<int>());
// dst = {9, 7, 5}
</int></int></int>
-
dst容量不足时,函数只填满dst.end(),不会越界——比手写循环加保护更可靠 - 源和目标可以是不同容器类型(比如
list→vector),但目标必须支持随机访问迭代器 - 如果
dst.size() > src.size(),多余位置保持默认构造(如int为 0),这点容易被忽略
比 partial_sort 更轻量的场景:nth_element
如果你只要前 N 个最大值的集合,**不要求它们内部有序**,nth_element 是更快的选择(平均 O(N)):
vector<int> v = {5, 2, 9, 1, 7, 3};
nth_element(v.begin(), v.begin() + 2, v.end(), greater<int>());
// 此时 v[0..2] 包含前 3 大的值,但顺序不确定;v[2] 是第 3 大的数
</int></int>
- 它只保证第 N 位(0-indexed)的元素就位,左边都 ≥ 它,右边都 ≤ 它
- 常用于快速找 Top-K 而不排序——比如排行榜实时计算,只需展示数值,不排具体名次
- 不能直接当
partial_sort替代品用;想取前 N 个再排序,还是得额外调一次sort(v.begin(), v.begin() + N, greater<int>())</int>
partial_sort 不做防护,段错误就在那儿等着。C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!











