347. 前 K 个高频元素

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

示例 1:

复制代码
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:

复制代码
输入: nums = [1], k = 1
输出: [1]

1、把容器内数都排列一遍

cpp 复制代码
class Solution {
public:
    struct cmp1{
        bool operator()(pair<int,int>&a,pair<int,int>&b){
            return a.second < b.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int>ump;
        vector<int>res(k);
        priority_queue<pair<int,int>,vector<pair<int,int>>,cmp1>pri_que;
         //用map统计出现频率
        for(int i = 0;i < nums.size();i++){
            ump[nums[i]]++;
        }

        //按value排序放入小根堆;
        for(auto it = ump.begin();it != ump.end();it++){
            pri_que.push(*it);
        }

        for(int i = 0;i < k;i++){
            res[i] = pri_que.top().first;
            pri_que.pop();
        }
        return res;
    }
};

二、优化;小根堆,只维护k个,其他的都pop掉,剩下的就是后面的较大者

cpp 复制代码
class Solution {
public:
    struct cmp1{
        bool operator()(pair<int,int>&a,pair<int,int>&b){
            return a.second > b.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int>ump;
        vector<int>res(k);
        priority_queue<pair<int,int>,vector<pair<int,int>>,cmp1>pri_que;
         //用map统计出现频率
        for(int i = 0;i < nums.size();i++){
            ump[nums[i]]++;
        }

        //按value排序放入小根堆;
        for(auto it = ump.begin();it != ump.end();it++){
            pri_que.push(*it);
            if(pri_que.size() > k){
                pri_que.pop();
            }
        }

        for(int i = 0;i < k;i++){
            res[i] = pri_que.top().first;
            pri_que.pop();
        }
        return res;
    }
};
相关推荐
LXS_35722 分钟前
Day 05 C++ 入门 之 指针
开发语言·c++·笔记·学习方法·改行学it
MicroTech20251 小时前
微算法科技(MLGO)研发突破性低复杂度CFG算法,成功缓解边缘分裂学习中的掉队者问题
科技·学习·算法
墨染点香1 小时前
LeetCode 刷题【126. 单词接龙 II】
算法·leetcode·职场和发展
aloha_7892 小时前
力扣hot100做题整理91-100
数据结构·算法·leetcode
Tiny番茄2 小时前
31.下一个排列
数据结构·python·算法·leetcode
挂科是不可能出现的2 小时前
最长连续序列
数据结构·c++·算法
前端小L3 小时前
动态规划的“数学之魂”:从DP推演到质因数分解——巧解「只有两个键的键盘」
算法·动态规划
RTC老炮3 小时前
webrtc弱网-ReceiveSideCongestionController类源码分析及算法原理
网络·算法·webrtc
mjhcsp3 小时前
C++ int 类型深度解析:从底层实现到实战应用
c++·int
21号 13 小时前
9.Redis 集群(重在理解)
数据库·redis·算法