关于priority_queue大小顶堆排序问题

在做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;
}

结果是升序:
在这里插入图片描述

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值