1.1我的想法
先用unordered_map统计每个数字出现的频率,以key作为数字,以value作为频率,对于value(频率)进行排序。最后将value相对应的数字,也就是key放入最终达到答案里面。
这里涉及到两个问题
1)是对于value进行排序,而map本身会对key进行排序,用sort也能对map的value进行排序
2)map不提供从value到key的查找
1.2 官方解答
class Solution {
public:
static bool cmp(pair<int, int>& m, pair<int, int>& n) {
return m.second > n.second;
}
//不明白为什么返回值要加上static
struct cmp{
bool operator()(pair<int,int>& m,pair<int,int>& n){
return m.second > n.second;
}
};
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int, int> occurrences;
for (auto& v : nums) {
//遍历容器的时候这里面加上引用是什么含义
//for(auto &a:b),循环体中修改a,b中对应内容也会修改
//for(auto a:b),循环体中修改a,b中内容不受影响
//for(const auto &a:b),a不可修改,用于只读取b中内容
occurrences[v]++;
}
//先用哈希表统计出每一个元素的出现的频率
// pair 的第一个元素代表数组的值,第二个元素代表了该值出现的次数
priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(&cmp)> q(cmp);
for (auto& [num, count] : occurrences) {
if (q.size() == k) {
if (q.top().second < count) {
q.pop();
q.emplace(num, count);
}
} else {
q.emplace(num, count);
}
}
vector<int> ret;
while (!q.empty()) {
ret.emplace_back(q.top().first);
q.pop();
}
return ret;
}
};
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/top-k-frequent-elements/solution/qian-k-ge-gao-pin-yuan-su-by-leetcode-solution/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
我的最终的解法
在设定优先队列的优先级的时候使用了functor.
class Solution{
public:
struct cmp{
bool operator()(pair<int,int>& m,pair<int,int>& n){
return m.second > n.second;
}
};
//class 里面还可以包含class吗?
//可以此时外界的class对于里面的class相当于一个namespace.
vector<int> topKFrequent(vector<int>& nums,int k){
unordered_map<int,int> occurrences;
for(auto& i:nums){
occurrences[i]++;
}
priority_queue<pair<int,int>,vector<pair<int,int>>,cmp> q;
for(auto&[num,count] : occurrences){
if(q.size() == k){
if(q.top().second < count){
q.pop();
q.emplace(num,count);
}
}
else q.emplace(num,count);
}
vector<int> ret;
while(!q.empty()){
ret.push_back(q.top().first);
q.pop();
}
return ret;
}
};
最终学会的写法,较为简洁
class Solution {
public:
struct cmp{
bool operator()(pair<int,int>& m,pair<int,int>& n){
return m.second > n.second;
}
};
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int,int> myMap;
for(auto& i:nums){
myMap[i]++;
}
priority_queue<pair<int,int>,vector<pair<int,int>>,cmp> que;
for(auto& p:myMap){
que.push(p);
if(que.size() > k)
que.pop();
}
vector<int> ret;
while(!que.empty()){
ret.push_back(que.top().first);
que.pop();
}
return ret;
}
};
1.4学到了什么
1)map容器第三个参数可放置一个比较函数,根据比较函数的规则对key进行排序
template < class Key, class T, class Compare = less<Key>,
class Allocator = allocator<pair<const Key,T> > > class map;
2)priority_queue优先队列
priority_queue
Container必须是用数组实现的容器,比如vector,deque等等,但不能用 list。STL里面默认用的是vector。
对于优先队列的优先级的设定不是很明白