LeetCode 347. Top K Frequent Elements 题解（C++）

LeetCode 347. Top K Frequent Elements 题解（C++）

---

题目描述

• Given a non-empty array of integers, return the k most frequent elements.

举例

• Given [1,1,1,2,2,3] and k = 2, return [1,2].

补充

• You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
• Your algorithm’s time complexity must be better than O(n log n), where n is the array’s size.

思路

• 对nums遍历，使用unordered_map< int, int>保存每个整数出现的次数，记做elementNums，第一个int表示数字，第二个整数表示该数字出现的次数；
• 使用优先队列priority_queue< pair< int, int> >，将上面保存的elementNums的第二个int（代表该数字出现的次数）和第一个int（表示数字）分别作为pair< int, int>的第一个参数和第二个参数，并入队列，若队列的长度大于elementNums的长度减去k，则将队列头的pair< int, int>的第二个int放到结果数组内，并将队列头出队列；
• 因为优先队列使用堆实现，所以该算法的时间复杂度为O（nlog(n-k))。

代码

class Solution 
{
public:
    vector<int> topKFrequent(vector<int>& nums, int k) 
    {
        unordered_map<int, int> elementNums;
        for (int num : nums)
        {
            elementNums[num]++;
        }

        priority_queue<pair<int, int> > pq; 
        vector<int> result;
        for (auto element : elementNums)
        {
            pq.push(pair<int, int> (element.second, element.first));
            if (pq.size() > elementNums.size() - k)
            {
                result.push_back(pq.top().second);
                pq.pop();
            }
        }

        return result;
    }
};