在做leetcode的347. Top K Frequent Elements时,用到了priority_queue这个容器适配器,这个容器默认排序是大顶堆,关于它的排序准则正如代码随想录里说的(对应题解链接):
那么,我们看下priority_queue的函数模板定义:
以及priority_queue的介绍:
注意标黄的这句话。
我们可以看见,所用的容器是vector(当然我们可以改成别的可以随机访问的容器如deque),没看源码实现,做一波合理的推测:
priortity_queue只有这三种操作:
top() 操作是不是暗示了排序后的“最大”元素是在最上面,pop也是对“最大”元素进行操作。同时因为vector的内存是连续,进行pop操作应该在空间的最后处操作,不然在空间的起始处复杂度会大大增大(其中的原因看别的文章),所以可以猜到,priortity_queue在vector的空间里是“升序”排序,在上面又说priortity_queue默认是降序,显然与我的猜测矛盾。
但我不这么认为,我认为这里的升降序的评判角度不同,我的角度是元素从容器内的排序来说,而所谓的降序是从堆的角度看,而堆是一个完全二叉树,又因为默认是大顶堆,所以最大的元素在是根节点,从这个角度看,确实是从根节点往叶节点“降序”。
最后贴出实现对比:
- 针对这种情况的,小顶堆:
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
auto cmp = [](const pair<int, int>& num1, const pair<int, int>& num2) {
return num1.second > num2.second;
};
priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(cmp)> que(cmp);
unordered_map<int, int> umap;
for (const auto& num : nums) ++umap[num];
auto it = umap.cbegin();
for (int i = 0; i < k && it != umap.end(); ++i) {
que.push(*it);
++it;
}
for (int i = k; i < nums.size() && it != umap.end(); ++i) {
auto front = que.top();
if (it->second > front.second) {
que.pop();
que.push({it->first, it->second});
}
// que.push(*it);
// que.pop();
++it;
}
vector<int> res;
for (int i = 0; i < k; ++i) {
res.push_back(que.top().first);
que.pop();
}
return res;
}
};
结果是降序:
- 以及sort的排序准则:
#include <iostream>
#include <vector>
#include <algorithm>
int main () {
std::vector<int> vec{1, 3, -2, 3, 5, 10};
sort(vec.begin(), vec.end(), [](int n1, int n2) {
return n1 > n2;
});
for_each(vec.begin(), vec.end(), [](int n) {
std::cout << n << " ";
});
std::cout << std::endl;
}
结果是升序: