前k个高频单词

leetcode第 692题

题目描述

给一非空的单词列表，返回前 k 个出现次数最多的单词。

返回的答案应该按单词出现频率由高到低排序。如果不同的单词有相同出现频率，按字母顺序排序。
来看输入输出：

出现前K个问题我们最先想到得就是堆，但是我们所使用的堆一般都是以数字为基准建堆，如果出现以单词个数或者是数字个数为基准如何建堆？
我们提供了一种key-value的形式建堆，以value为基准值，以key为关键字。

pair<string , int> or pair<int , int>

在C++中有优先级队列这种数据结构，其底层是一个堆，我们可以通过仿函数来对这个堆进行想要的调整。

class Compare()
{	
	public:
	bool operator()
	{
	}
}
priority

有了这几种结构，我们可以先将单词个数统计出来，然后用堆进行调整。

class Compare{
public:
    bool operator()(const pair<string,size_t> &left, const pair<string,size_t> &right)
    {
         if(left.second == right.second)
        {
          int ret =  strcmp(left.first.c_str(),right.first.c_str());
            if(ret < 0)
            {
                return false;
            }
            if(ret >= 0)
            return true; 
        }
        return left.second < right.second;
    }
};
class Solution {
    public:
        vector<string> topKFrequent(vector<string>& words, int k) {
        map<string ,size_t> M;
        for(auto & it: words)
        {
            M[it]++;
        }
        priority_queue<pair<string,size_t> , vector<pair<string,size_t>> ,Compare> P;
        for(auto &e : M)
        {
            P.push(e);
        }
        vector<string> temp;
            for(int i = 0;i < k && P.size() > 0;++i)
            {
                string t = P.top().first;
                temp.push_back(t);
                P.pop();
            }
            return temp;
        }
};

leetcode 347题

前k个高频元素

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

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

同理：建立 key-value为<int ,int>的映射关系

class Compare
{
    public:
    bool operator() (const pair<int,int> &left , const pair<int,int> &right)
    {
        return left.second < right.second;
    }
};
class Solution {
public:  
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int, int> M;
        for (auto it : nums)
        {
            M[it]++;
        }
        priority_queue<pair<int, int>, vector<pair<int, int>> , Compare> P;
        for (auto it : M)
        {
            P.push(it);
        }
        vector<int> vec;
        for (int i = 0; i < k; ++i)
        {
            if (!P.empty())
            {
                vec.push_back(P.top().first);
                P.pop();
            }
        }
        return vec;
    }
};