优先队列(内部的一个排序,它本质是一个堆实现的)priority_queue<Type, Container, Functional>
Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式。当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆
//升序队列,小顶堆
priority_queue <int,vector<int>,greater<int> > q;
//降序队列,大顶堆
priority_queue <int,vector<int>,less<int> >q;
//greater和less是std实现的两个仿函数(就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator(),这个类就有了类似函数的行为,就是一个仿函数类了)
最小k个数
vector<int> getLeastNumbers(vector<int>& arr, int k)
{
vector<int> ret;
priority_queue<int, vector<int>, less<int>> q;
for (int i = 0; i < arr.size() - 1; i++)
{
q.push(arr[i]);
if (q.size() > k)
{
q.pop();
}
}
while (!q.empty())
{
int tmp = q.top();
ret.push_back(tmp);
q.pop();
}
return ret;
}
前k个高频数
给定一个数组,返回前k个出现频率最高的数字
class Solution1
{
public:
//定义一个仿函数,用于小顶堆
class comparison
{
public:
bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs)
{
return lhs.second > rhs.second;
}
};
vector<int> topKNum(vector<int>& nums, int k)
{
unordered_map<int, int> numap;
for (int i = 0; i < nums.size(); i++)
{
numap[nums[i]]++;
}
//用优先队列定义一个小根堆,大小为k,对频率排序
priority_queue<pair<int, int>, vector<pair<int, int>>, comparison> pri_que;
for (auto it = numap.begin(); it != numap.end(); it++)
{
pri_que.push(*it);
if (pri_que.size() > k)
{ // 如果堆的大小大于了K,则队列弹出,保证堆的大小一直为k
pri_que.pop();
}
}
vector<int> result;
for (int i = k - 1; i >= 0; i--)
{
result.push_back(pri_que.top().first);
pri_que.pop();
}
return result;
}
};