力扣刷题之前K个高频元素刷题知识学习

题目描述

给定一个非空的整数数组,返回其中出现频率前 k 高的元素。

示例 1:

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

输入: nums = [1], k = 1 输出: [1] 链接:https://leetcode-cn.com/problems/top-k-frequent-elements

题解

`class Solution { public:

// 小顶堆 class mycomparison { 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; // map<nums[i],对应出现的次数>
    for (int i = 0; i < nums.size(); i++)
        map[nums[i]]++;
    // 对频率排序
    // 定义一个小顶堆,大小为k
    //定义:priority_queue<Type, Container, Functional>

//Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式。 priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparison> pri_que;

    for(unordered_map<int,int>::iterator it=map.begin();it!=map.end();it++)
    {
        pri_que.push(*it);
        if(pri_que.size()>k)
            pri_que.pop();
    }
    vector<int> res(k);
    for(int i=k-1;i>=0;i--)
    {
        res[i]=pri_que.top().first;
        pri_que.pop();
    }
    return res;
}

};`

优先队列知识介绍

普通的队列是一种先进先出的数据结构,元素在队列尾追加,而从队列头删除。

在优先队列中,元素被赋予优先级。当访问元素时,具有最高优先级的元素最先删除。优先队列具有最高级先出 (first in, largest out)的行为特征。

首先要包含头文件#include, 他和queue不同的就在于我们可以自定义其中数据的优先级, 让优先级高的排在队列前面,优先出队。

优先队列具有队列的所有特性,包括队列的基本操作,只是在这基础上添加了内部的一个排序,它本质是一个堆实现的。

和队列基本操作相同:

top 访问队头元素 empty 队列是否为空 size 返回队列内元素个数 push 插入元素到队尾 (并排序) emplace 原地构造一个元素并插入队列 pop 弹出队头元素 swap 交换内容 定义:priority_queue<Type, Container, Functional> Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式。

当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆。 一般是:

//升序队列

小顶堆 priority_queue <int,vector,greater > q;

//降序队列

大顶堆 priority_queue <int,vector,less >q;

//greater和less是std实现的两个仿函数(就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator(),这个类就有了类似函数的行为,就是一个仿函数类了)

posted @ 2021-02-01 12:11  justloving  阅读(138)  评论(0编辑  收藏  举报