692. 前K个高频单词

最新推荐文章于 2024-11-12 13:55:23 发布

Bloo8m

最新推荐文章于 2024-11-12 13:55:23 发布

阅读量343

点赞数 5

文章标签： c++

本文链接：https://blog.csdn.net/qq_51791303/article/details/137611228

版权

本文讨论了在C++中使用`priority_queue`时遇到的问题，作者自定义了一个排序函数，但在实际操作中队列元素顺序与预期相反。作者推测可能是因为`priority_queue`内部的优化导致。

摘要由CSDN通过智能技术生成

题目
本题熟悉下priority_queue,感觉有点怪，自定义排序和greater<>,less<>不太一样。
因为我这里写的排序方式是：

   auto cmp=[&](string a,string b){
            return mm[a]==mm[b]? a<b:mm[a]>mm[b];
        };

但是我遍历queue:

    while(!q.empty()){
        cout<<q.top()<<endl;
        q.pop();
    }

发现结果是：
在这里插入图片描述
这和我排序的预期刚好相反，感觉应该是peiorityqueue内部的问题，因为虽然顺序是反的，但我想用堆来解决，而这个顺序方便我后续的堆操作。可能是内部自己优化了吧，我预期结果应该是 i love coding leetcode

1.本题注意下priority_queue的自定义写法。

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
    unordered_map<string,int> mm;
    for(auto i: words) mm[i]++;
    
    auto cmp=[&](string a,string b){ return mm[a]==mm[b]? a<b:mm[a]>mm[b];};    //cmp可以用auto，之前都是匿名写法

    priority_queue<string,vector<string>,decltype(cmp)> q(cmp);   // 注意写法
    
    for(auto i=mm.begin();i!=mm.end();i++) q.push(i->first);
    vector<string> ans(k);
    while(!q.empty()){
        ans[--k]=q.top();                   //这个地方注意下，得倒着写入vector
        q.pop();
    }
    return ans;    
    }
};