Leetcode347前 K 个高频元素 -hot100-代码随想录

meeiuliuus

已于 2024-05-04 17:33:11 修改

阅读量402

点赞数 8

分类专栏： # leetcode ---medium 文章标签：算法 c++ leetcode

于 2024-01-22 12:10:50 首次发布

本文链接：https://blog.csdn.net/qq_52313711/article/details/135745013

版权

leetcode ---medium 专栏收录该内容

172 篇文章 0 订阅

订阅专栏

本文介绍了如何使用C++实现一个名为`topKFrequent`的函数，利用优先队列（大根堆）和自定义比较器，找出给定整数数组中出现频率最高的k个元素。作者逐步解决了从首刷到bug-free版本的代码优化过程。

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

题目：

代码(首刷看解析 2024年1月22日）：

class Solution {
public:
    class Mycomparison{
    public:
        bool operator()(const pair<int,int>& lhs,const pair<int,int>& rhs) {
            return lhs.second > rhs.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> map;
        for (int i = 0; i < nums.size(); ++i) {
            map[nums[i]]++;
        }

        priority_queue<pair<int,int>,vector<pair<int,int>>,Mycomparison> pri_que;
        for (unordered_map<int, int>::iterator it = map.begin(); it != map.end(); it++) {
            pri_que.push(*it);
            if (pri_que.size() > k) {
                pri_que.pop();
            }
        }
        vector<int> result(k);
        for (int i = k - 1; i >= 0; i--) {
            result[i] = pri_que.top().first;
            pri_que.pop();
        }
        return result;
    }
};

优先队列定义方法搞清楚~，尤其是第三个参数的自定义

代码（二刷debug 看解析 2024年3月9日）

我就不太懂，为什么在往q里插入元素的时候要限定q的大小一定==k，不限定就会出错。按道理说priority_queue的大根堆pop之后应该会在内部调整结构的呀，为什么我把18行的if去掉后就不能通过了呢？

class Solution {
public:
    struct cmp{
        bool operator()(pair<int,int>& a, pair<int,int>& b) {
            return a.second > b.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        int n = nums.size();
        unordered_map<int, int> um;
        for (int i = 0; i < nums.size(); ++i) {
            um[nums[i]]++;
        }

        priority_queue<pair<int,int>, vector<pair<int,int>>, cmp> q;
        for (auto it : um) {
            if (q.size() == k) {
                if (q.top().second < it.second) {
                    q.pop();
                    q.push(it);
                }
            } else {
                q.push(it);
            }
            
        }

        vector<int> result(k);
        for (int i = 0; i < k; ++i) {
            result[i] = q.top().first;
            q.pop();
        }

        return result;
    }
};

代码（三刷自解 2024年5月4日 9min bugfree）

class Solution {
public:
    //优先队列
    struct cmp{
        bool operator()(pair<int,int>& p1,pair<int,int>& p2) {
            return p1.second < p2.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> um;
        for (int x : nums) {
            um[x]++;
        }
        priority_queue<pair<int,int>,vector<pair<int,int>>, cmp> pq;
        for (const auto& kv : um) {
            pq.push(kv);
        }
        vector<int> res;
        while(k--) {
            res.push_back(pq.top().first);
            pq.pop();
        }
        return res;
    }
};