leetcode347. 前 K 个高频元素

题目描述

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

示例 1:

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

示例 2:

输入: nums = [1], k = 1
输出: [1]

提示：

你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
你可以按任意顺序返回答案。

code

• 求top k大，使用堆排序-小根堆

 class Solution {
public:
   //小根堆使用>!!!!!!!
   struct cmp{
       bool operator()(const pair<int,int>& a,const pair<int,int>& b)
       {
           return a.second>b.second;
       }
   };
   vector<int> topKFrequent(vector<int>& nums, int k) {
       //构建小根堆
       priority_queue<pair<int,int>,vector<pair<int,int> >,cmp> minHeap;
       //map存放频率
       map<int,int> counter;
       for(auto num : nums){
           counter[num]++;
       }
       //将top k频率的数存入minHeap中
       for(auto pair : counter){
           //auto pair = make_pair(p.first,p.second);
           if(minHeap.size()<k){
               minHeap.push(pair);
               //printf("(%d,%d)\n",pair.first,pair.second);
           }
           else if(pair.second>=minHeap.top().second){//必须是大于等于
               minHeap.pop();
               minHeap.push(pair);
               //printf("(%d,%d)\n",pair.first,pair.second);
           }
       }

       //从小根堆中取出数据
       vector<int> res(minHeap.size());
       while(!minHeap.empty()){
           res[--k] = minHeap.top().first;
           minHeap.pop();
       }

       return res;
   }
};

补充

• top k问题：参考前k小、前k大算法：快排、堆排序、冒泡，一般用堆排序
  • 求前 k大~ 用小根堆;
    求前 k小~ 用大根堆
    (求前k大，就维持一个k大小的小根堆，求前k小，就维持一个k大小的大根堆)
  • 一篇理解堆排序的优秀博文：堆排序算法（图解详细流程）
• c++使用优先队列priority_queue实现（注意默认是大根堆）

    priority_queue<int, vector<int>, less<int>>s;   //less降序，大根堆
  
    priority_queue<int, vector<int>, greater<int>>s; //greater升序，小根堆
  

• pair实际上是一个含有两个元素first和second的结构体，默认情况下按first排序，要想按second排序，可考虑颠倒first和second，或者自己写排序函数cmp
• auto是C++11版本及之后的特性

? 小根堆明明是升序，为什么排序规则里要写成降序嘞？？

---

参考链接

• 使用C++优先队列（priority_queue）解决Top K问题
• STL-priority_queue用法（重点: 升序，小根堆）
• 关于map中的pair的用法