【每日刷题】前K个高频元素

day24, 前K个高频元素

题目来源：leetcode
给定一个非空的整数数组，返回其中出现频率前 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 是数组的大小。

解答：首先使用map遍历数组形成一一映射关系，其中key为元素，value为元素出现的频率。
随后建立pair数组，按照value, key 的顺序存放到pair数组中，对该数组排序，选择前k个元素即可。

代码：

vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int,int> record;
        vector< pair<int, int > > record2;
        vector<int> result;
        
        for( int i = 0; i < nums.size(); i++)
            record[ nums[i]]++;
        
        for( auto it = record.begin(); it != record.end(); it++)
            record2.push_back( pair<int, int>( it->second, it->first));
        
        sort( record2.begin(), record2.end()); 
        
        for( int i = record2.size() - 1; i >=0 && k > 0; i--, k--)
            result.push_back( record2[i].second);
        
        return result;
    }

运行结果：

---

我的微信公众号