力扣第347题前K个高频元素

news2024/12/22 19:31:56

前言

记录一下刷题历程力扣第347题前K个高频元素

前K个高频元素

原题目：

分析

我们首先使用哈希表来统计数字出现的频率，然后我们使用一个桶排序。我们首先定义一个长度为n+1的数组，对于下图这个示例就是长度为7的数组。为什么需要一个长度为n+1的数组呢？假如说总共有三个数字都为1，那么我们需要把这个1放在数组下标为3的位置，假如说数组长度为n，对于这个例子就是长度为3，那么它的数组下标最大为2，不能存储在下标为3 的位置，所以我们需要创建一个长度为n+1的数组。根据我们的哈希表数字1它出现了三次那么我们把它放在数组下标为3的位置，同理数字2出现了两次我们把它放在数组下标为2的位置，最后是数字3它出现了1次所以我们把它放在数组下标为1的位置，假如说还有一个数字4他也出现了三次，那么我们将它和1共同存储在下标为3的位置，最后我们从后往前遍历数组即可。
在这里插入图片描述

代码如下：

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        vector<int> res; // 用于存放结果的数组
        unordered_map<int, int> u_mapCount; // 用于记录每个数字的出现次数
        vector<vector<int>> Freq(nums.size() + 1, vector<int>()); // 用于记录频次对应的数字

        // 统计每个数字出现的频率
        for (int i = 0; i < nums.size(); i++) {
            u_mapCount[nums[i]] += 1;
        }

        // 将每个数字根据其频率放入对应的频率桶中
        for (auto& pair : u_mapCount) {
            Freq[pair.second].emplace_back(pair.first);
        }

        // 从高到低遍历频率桶，收集出现频率最高的数字
        for (int i = Freq.size() - 1; i >= 0; i--) {
            for (int j = 0; j < Freq[i].size(); j++) {
                res.emplace_back(Freq[i][j]);
                if (res.size() == k) { // 如果结果数组已包含k个元素，返回结果
                    return res;
                }
            }
        }
        
        // 返回结果（实际不应到达这里，因为k通常会小于结果数组的大小）
        return res;
    }
};