LintCode 550. 最常使用的K个单词II（自定义set（可修改数据的优先队列） + map）

样例 1:
输入：
TopK(2)
add("lint")
add("code")
add("code")
topk()
输出：["code", "lint"]
解释：
"code" 出现两次并且 "lint" 出现一次， 它们是出现最频繁的两个单词。

样例 2:
输入：
TopK(1)
add("aa")
add("ab")
topk()
输出：["aa"]
解释：
"aa" 和 "ab" 出现 , 但是aa的字典序小于ab。

注意事项
如果两个单词有相同的使用频率, 按字典序排名.

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，不能用count
            q.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());
    }
};