[leetcode]347. Top K Frequent Elements

 1 class CompareDist
 2 {
 3 public:
 4     bool operator()(pair<int, int> n1, pair<int, int> n2) 
 5     {
 6         return n1.second > n2.second;
 7     }
 8 };
 9 
10 class Solution {
11 public:
12     vector<int> topKFrequent(vector<int>& nums, int k) 
13     {
14         vector<int> ret;
15         unordered_map<int, int> htable;
16         
17         for (int key : nums) // get each key appears times
18             htable[key]++;
19 
20         priority_queue<pair<int, int>, vector<pair<int, int> >, CompareDist> sheap; // use min heap to get k biggest
21         for (auto elem : htable)
22         {
23             if (sheap.size() < k)
24             {
25                 sheap.push(elem);
26             }
27             else
28             {
29                 pair<int, int> heap_min = sheap.top();
30                 if (elem.second > heap_min.second)
31                 {
32                     sheap.pop();
33                     sheap.push(elem);
34                 }
35             }
36         }
37         
38         while (!sheap.empty())
39         {
40             pair<int, int> heap_min = sheap.top();
41             ret.push_back(heap_min.first);
42             sheap.pop();
43         }
44         reverse(ret.begin(), ret.end());
45         
46         return ret;
47     }
48 };