算法(栈与队列)——前K个高频元素

文章目录

• • 1.题目描述
  • 2.解题思路
  • 3.代码实现

1.题目描述

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

示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]

示例 2:
输入: nums = [1], k = 1
输出: [1]

2.解题思路

    这一题我们需要使用小顶堆这种数据结构，和队列是类似的，只不过小顶堆内部的元素是按由小到大顺序排列的。我们需要定义一个unordered_map来储存每种元素的对应数量，然后根据map[nums[i]].second的大小，将其写入int result里，并且注意写入时的顺序。

3.代码实现

class Solution {
public:
class MyComparision{
public:
    bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs)
    {
        return lhs.second > rhs.second;
    }
};
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> map;
        for(int i = 0; i < nums.size(); i++)//记录个元素出现的次数
        {
            map[nums[i]]++;
        }
        priority_queue<pair<int, int>, vector<pair<int, int>>, MyComparision> pri_que;
        //定义一个小顶堆，大小为k
        for(unordered_map<int, int>::iterator it = map.begin(); it != map.end(); it++)
        {
            pri_que.push(*it);
            if(pri_que.size() > k)//小顶堆中的元素数量大于k时，将频率小的元素弹出
            {
                pri_que.pop();
            }
        }
        vector<int> res(k);
        for(int j = k - 1; j >= 0; j--)
        {
            res[j] = pri_que.top().first;
            pri_que.pop();
        }
        return res;
    }
};