347.前 K 个高频元素

题目描述

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

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

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

解题思路

使用一个hash_map统计给定数组中各值的出现次数，然后按出现次数排序，返回topK的对应值。
思路比较简单，c++实现上很有技巧。

    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int, int> counter;
        for(auto val : nums)
            counter[val]++;
        
        vector<pair<int, int> > counter_list(counter.begin(), counter.end());
        std::sort(counter_list.begin(), counter_list.end(), [](const pair<int, int>& l, const pair<int, int>& r){
            return l.second>r.second;
        });
        
        vector<int> ret;
        for(int i=0; i<k; i++)
            ret.push_back(counter_list[i].first);

        return ret;
    }