力扣.----前K个高频元素

题目

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

示例 1:

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

示例 2:

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

提示：

• 1 <= nums.length <= 105
• k 的取值范围是 [1, 数组中不相同的元素的个数]
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

代码

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
      unordered_map<int, int> freqMap;
    
    // 统计每个数字出现的频率
    for (int num : nums) {
        freqMap[num]++;
    }
    
    // 定义一个自定义排序规则，按照出现频率降序排列
    auto cmp = [](pair<int, int>& a, pair<int, int>& b) {
        return a.second > b.second;
    };
    
    // 将频率信息存入 vector 中，方便排序
    vector<pair<int, int>> freqVec(freqMap.begin(), freqMap.end());
    
    // 对频率进行排序
    sort(freqVec.begin(), freqVec.end(), cmp);
    
    // 提取前 k 个高频元素
    vector<int> result;
    for (int i = 0; i < k; ++i) {
        result.push_back(freqVec[i].first);
    }
    
    return result;
    }
};