LeetCode 347. 前K个高频元素(C++)

题目：

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

输入: nums = [1], k = 1
输出: [1]

说明：

你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。

思路：

由于在本题中使用排序算法的复杂度不能满足复杂度必须优于 O(n log n)的要求，所以考虑采用优先队列实现。

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> freq;    //(元素，频率)
        for(int i = 0; i < nums.size(); i++){
            freq[nums[i]]++;
        }

        priority_queue< pair<int,int>, vector<pair<int,int>>, greater<pair<int,int>> > pq;    //(频率，元素)

        for(auto a : freq){
            if(pq.size() == k){
                if(a.second > pq.top().first){
                    pq.pop();
                    pq.push(make_pair(a.second,a.first));
                }
            }
            else{
                 pq.push(make_pair(a.second,a.first));
            } 

        }
        vector<int> res;
        while(!pq.empty()){
            res.push_back(pq.top().second);
            pq.pop();
        }
        return res;
    }
};

问题待解决：

采用下面的代码遍历unordered_map最后的输出结果会出错，现在还没有找到问题原因。

        for(auto iter; iter != freq.end(); iter++){
            if(pq.size() == k){  //优先队列中已经有k个元素及其频率
               if(iter->second > pq.top().first);{    //遍历到的元素的出现频率大于优先队列中元素出现的最小频率
                    pq.pop();
                    pq.push(make_pair(iter->second,iter->first));
               }
            }
            else{
                 pq.push(make_pair(iter->second,iter->first));
            }

        }