看到这个题首线就想到 将所有的数放在map里面问题在于怎么拿出来,想了很久都没想到,最后看了别人的,才知道用优先队列的方式
复习一下
优先队列的定义方式 :priority_queue<Type, Container, Functional>
Type 就是数据类型,Container 就是容器类型(Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector),Functional 就是比较的方式,当需要用自定义的数据类型时才需要传入这三个参数,使用基本数据类型时,只需要传入数据类型,默认是大顶堆
这个Fun 有两个仿函数 分别是
greater<>升序
less<>降序
这这里还看到别人用到了using 关键字
这个关键字相当于 typedef
在继承中可以将私有继承的函数作为自己的公共成员函数
AC 代码:
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
map<int ,int>M;
using pii = pair<int,int>;
priority_queue<pii,vector<pii>,greater<pii>>pq;
vector<int>ans;
for(int i = 0;i<nums.size();++i)
{
if(M.find(nums[i]) == M.end())
{
M.insert(make_pair(nums[i],1));
}
else
{
int t = M[nums[i]] + 1;
M[nums[i]]+= 1;
}
}
for(auto it : M)
{
pq.emplace(it.second,it.first);
if(pq.size() > k)pq.pop();
}
while(!pq.empty())
{
ans.emplace_back(pq.top().second);
pq.pop();
}
return ans;
}
};