LeetCode-347：前K个高频元素

最新推荐文章于 2024-08-11 23:49:29 发布

Wade_Gao

最新推荐文章于 2024-08-11 23:49:29 发布

阅读量467

点赞数

分类专栏： LeetCode 堆文章标签：数据结构 leetcode

本文链接：https://blog.csdn.net/weixin_44587168/article/details/105377238

版权

LeetCode 同时被 2 个专栏收录

63 篇文章 0 订阅

订阅专栏

堆

7 篇文章 0 订阅

订阅专栏

一、题目描述

在这里插入图片描述

二、解题思路

使用优先队列，设置为小根堆

首先用map记录数据出现的频度
维护一个大小为K的小根堆
遍历map，若当前遍历到的map元素的出现频度小于小根堆堆顶元素，即若该元素的出现频度比当前小根堆最小的还小，丢弃不管，否则将小根堆堆顶元素pop掉，将该元素压入堆，再调整堆

三、解题代码

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int, int> counter;
        priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int,int>> > Heap;
        for(auto e: nums)
            counter[e]++;			//对map的随机访问
        for(auto &x: counter){
            auto pr = make_pair(x.second, x.first);
            if(Heap.size() == k){
                if(pr < Heap.top())    continue;
                else    Heap.pop();
            }
            Heap.push(move(pr));	//不进行数据拷贝，只将内存控制权转移
        }
        vector<int> sln(Heap.size());
        k = sln.size();
        while(!Heap.empty()){
            sln[--k] = Heap.top().second;
            Heap.pop();
        }
        return sln;
    }
};