LeetCode:347.前K个高频元素

方法 1:用 vector 存储键值对,按频率排序(简单直接)
cpp 复制代码
class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        // 1. 统计每个元素的频率
        unordered_map<int, int> freq;
        for (int num : nums) {
            freq[num]++;
        }

        // 2. 将键值对存入vector,便于按频率排序
        vector<pair<int, int>> freq_vec(freq.begin(), freq.end());

        // 3. 按频率降序排序(第二个元素是频率)
        sort(freq_vec.begin(), freq_vec.end(), 
            [](const pair<int, int>& a, const pair<int, int>& b) {
                return a.second > b.second; // 频率高的排在前
            });

        // 4. 取前k个元素的键
        vector<int> result;
        for (int i = 0; i < k; ++i) {
            result.push_back(freq_vec[i].first);
        }

        return result;
    }
};
进阶方法 2:用优先队列(堆)优化(时间复杂度 O (n log k))
cpp 复制代码
class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        // 1. 统计频率
        unordered_map<int, int> freq;
        for (int num : nums) {
            freq[num]++;
        }

        // 2. 用小顶堆存储前k个高频元素(堆顶是最小的高频元素)
        // 堆中元素是 (频率, 元素),按频率升序排列
        priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;

        for (auto& pair : freq) {
            pq.push({pair.second, pair.first});
            // 当堆大小超过k,弹出最小的高频元素
            if (pq.size() > k) {
                pq.pop();
            }
        }

        // 3. 从堆中取出结果(此时堆中是前k个高频元素)
        vector<int> result;
        while (!pq.empty()) {
            result.push_back(pq.top().second);
            pq.pop();
        }
        // 由于堆是小顶堆,结果是从小到大的,如需保持高频在前可反转
        reverse(result.begin(), result.end());

        return result;
    }
};

用小顶堆只保留前 k 个高频元素,每次插入后若超过 k 就弹出最小的,效率更高,尤其当 n 很大而 k 较小时。

相关推荐
Greedy Alg5 小时前
LeetCode 239. 滑动窗口最大值
数据结构·算法·leetcode
空白到白6 小时前
机器学习-KNN算法
人工智能·算法·机器学习
闪电麦坤956 小时前
数据结构:排序算法的评判标准(Criteria Used For Analysing Sorts)
数据结构·算法·排序算法
爱coding的橙子6 小时前
每日算法刷题Day65:8.27:leetcode dfs11道题,用时2h30min
算法·leetcode·深度优先
不懂机器人6 小时前
linux网络编程-----TCP服务端并发模型(epoll)
linux·网络·tcp/ip·算法
Mercury_Lc7 小时前
【链表 - LeetCode】25. K 个一组翻转链表
数据结构·leetcode·链表
地平线开发者7 小时前
理想汽车智驾方案介绍 3|MoE+Sparse Attention 高效结构解析
算法·自动驾驶
好学且牛逼的马8 小时前
golang 10指针
开发语言·c++·golang
小O的算法实验室9 小时前
2025年KBS SCI1区TOP,矩阵差分进化算法+移动网络视觉覆盖无人机轨迹优化,深度解析+性能实测
算法·论文复现·智能算法改进
艾莉丝努力练剑10 小时前
【C语言16天强化训练】从基础入门到进阶:Day 11
c语言·学习·算法