leetcode692 Top K Frequent Words

1 #include <bits/stdc++.h>
2 using namespace std;
3
4 class Solution
5 {
6 public:
7     inline bool cmp(pair<int, string> a, pair<int, string> b)
8     {
9         if (a.first > b.first) return true;
10         if (a.first == b.first && a.second < b.second) return true;
11         return false;
12     }
13     struct cmp1
14     {
15         bool operator()(const pair<int, string>& a, const pair<int, string>& b) const
16         {
17             if (a.first > b.first) return true;
18             if (a.first == b.first && a.second < b.second) return true;
19             return false;
20         }
21     };
22     vector<string> topKFrequent(vector<string>& words, int k)
23     {
24         unordered_map<string, int> mp;
25         for (auto it : words) mp[it]++;
26         priority_queue<pair<int, string>, vector<pair<int, string>>, cmp1> q;
27         int cnt = 0;
28         for (auto it : mp)
29         {
30             pair<int, string> tmp(it.second, it.first);
31             if (cnt < k)
32             {
33                 q.push(tmp);
34             }
35             else if (cmp(tmp, q.top()))
36             {
37                 q.push(tmp);
38                 q.pop();
39             }
40             cnt++;
41         }
42         vector<string> ret;
43         while (!q.empty())
44         {
45             ret.push_back(q.top().second); q.pop();
46         }
47         reverse(ret.begin(), ret.end());
48         return ret;
49     }
50 };
51 int main()
52 {
53     vector<string> v = {"i", "love", "leetcode", "i", "love", "coding"};
54     vector<string> ret = Solution().topKFrequent(v, 2);
55     for (auto it : ret)
56     {
57         cout << it << " ";
58     }
59     cout << endl;
60     return 0;
61 }