1. 題目

在實時數(shù)據(jù)流中找到最常使用的k個單詞.
實現(xiàn)TopK類中的三個方法:

• TopK(k), 構造方法
• add(word), 增加一個新單詞
• topk(), 得到當前最常使用的k個單詞.

樣例 1: 輸入： TopK(2) add("lint") add("code") add("code") topk() 輸出：["code", "lint"] 解釋： "code" 出現(xiàn)兩次并且 "lint" 出現(xiàn)一次， 它們是出現(xiàn)最頻繁的兩個單詞。樣例 2: 輸入： TopK(1) add("aa") add("ab") topk() 輸出：["aa"] 解釋： "aa" 和 "ab" 出現(xiàn) , 但是aa的字典序小于ab。注意事項 如果兩個單詞有相同的使用頻率, 按字典序排名.

2. 解題

• 優(yōu)先隊列，修改內(nèi)部數(shù)據(jù)很麻煩，利用set，自定義其排序規(guī)則
• 遇到要更新的數(shù)據(jù)，先刪除舊的數(shù)據(jù)，再插入更新的
• 遇到兩點需要注意的，比較操作，必須const，查找存在，不能count，可能是因為自定義 set 的原因。

unordered_map<string, int> wc; struct cmp {bool operator()(const string &a,const string &b) { //必須寫const，不寫報錯if (wc[a] == wc[b])return a < b;return wc[a] > wc[b];} };class TopK { private:set<string,cmp> q;int K; public:TopK(int k) { K = k;}void add(string word) {if (q.find(word) != q.end())//必須用find，不能用countq.erase(word);wc[word]++;q.insert(word);if (q.size() > K)q.erase(--q.end());}vector<string> topk() {return vector<string> (q.begin(), q.end());} };

100% 數(shù)據(jù)通過測試
總耗時 1314 ms
您的提交打敗了 37.80% 的提交!

總結

以上是生活随笔為你收集整理的LintCode 550. 最常使用的K个单词II（自定义set（可修改数据的优先队列） + map）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。