【L347】Top-K问题

描述

【Leet code 347】给定一个数组，返回前K个出现次数最多的元素。

思路

1.构建哈希表(std::unordered_map)记录元素和出现次数
2.构建优先队列(std::priority_queue)遍历哈希表入队
3.优先队列出队k个元素

【注意】优先队列的cmp函数和快排形式相反，想要得到大顶堆，left<right

复杂度

时间：O(N*log(k))
空间：O(N)

代码（C++）

/* 【L 347】Top-K 问题
* 返回前K个元素
*/
class cmp_prt_queue
{
public:
    bool operator()(std::pair<int, int>& v1, std::pair<int, int>& v2)
    {
        if (v1.second < v2.second) return true;
        return false;
    }
};
/*
priority_queue<Type, Container, Functional>
Type 就是数据类型，Container 就是容器类型（Container必须是用数组实现的容器，比如vector, deque等等，但不能用 list。STL里面默认用的是vector），Functional 就是比较的方式，当需要用自定义的数据类型时才需要传入这三个参数，使用基本数据类型时，只需要传入数据类型，默认是大顶堆
*/
class Solution {
public:
    std::vector<int> topKFrequent(std::vector<int>& src, int k)
    {
        std::vector<int> topk;
        // 建立一个哈希表记录元素出现次数
        std::unordered_map<int, int> temp_map;
        for (auto& ele : src)
        {
            if (temp_map.find(ele) == temp_map.end())
            {
                temp_map[ele] = 1;
            }
            else
            {
                temp_map[ele] ++;
            }
        }

        // 优先队列(<k,v>, vector<k,v>, cmp)
        std::priority_queue<std::pair<int, int>, std::vector<std::pair<int, int>>, cmp_prt_queue> prt_queue;
        for (auto iter : temp_map)
        {
            //std::cout << iter.first << "," << iter.second << std::endl;
            prt_queue.push(std::make_pair(iter.first, iter.second));
        }

        for (int idx = 0; idx < k; idx++)
        {
            topk.push_back(prt_queue.top().first);
            prt_queue.pop();
        }
        return topk;
    }
};