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;
    }
};
相关推荐
·白小白12 小时前
力扣(LeetCode) ——11.盛水最多的容器(C++)
c++·算法·leetcode
深思慎考13 小时前
RabbitMQ 入门:基于 AMQP-CPP 的 C++ 实践指南与二次封装
开发语言·c++·分布式·rabbitmq·api
深思慎考13 小时前
Ubuntu 系统 RabbitMQ 安装指南与使用(含 C++ 客户端与 SSL 错误解决)
c++·ubuntu·rabbitmq·github·rabbitmqpp
沐浴露z13 小时前
【JVM】详解 垃圾回收
java·jvm·算法
代码欢乐豆13 小时前
编译原理机测客观题(7)优化和代码生成练习题
数据结构·算法·编译原理
草莓熊Lotso14 小时前
《C++ Web 自动化测试实战:常用函数全解析与场景化应用指南》
前端·c++·python·dubbo
东巴图14 小时前
分解如何利用c++修复小程序的BUG
开发语言·c++·bug
祁同伟.14 小时前
【C++】二叉搜索树(图码详解)
开发语言·数据结构·c++·容器·stl
恒者走天下14 小时前
AI智能网络检测项目(cpp c++项目)更新
开发语言·c++
Scc_hy14 小时前
强化学习_Paper_2000_Eligibility Traces for Off-Policy Policy Evaluation
人工智能·深度学习·算法·强化学习·rl