【LeetCode】—— 前k个高频单词和元素

一、前k个高频单词LeetCode692

1.1 题目描述

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

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

示例 1：

输入: [“i”, “love”, “leetcode”, “i”, “love”, “coding”], k = 2
输出: [“i”, “love”] 解析: “i” 和 “love” 为出现次数最多的两个单词，均为2次。
注意，按字母顺序 “i” 在 “love” 之前。

示例 2：

输入: [“the”, “day”, “is”, “sunny”, “the”, “the”, “the”, “sunny”, “is”, “is”], k = 4
输出: [“the”, “is”, “sunny”, “day”]
解析: “the”, “is”, “sunny” 和 “day” 是出现次数最多的四个单词
出现次数依次为 4, 3, 2 和 1 次。

注意：

假定 k 总为有效值， 1 ≤ k ≤ 集合元素数。
输入的单词均由小写字母组成。

1.2 解题思路

C++实现，这里使用STL中的map，键值对的思想在这很适用

• 1、首先统计每个单词出现的次数，定义一个countmap，此时key为单词，value单词出现的次数，此时单词已经按照字母顺序被排序
• 2、由于出现的次数可能重复，因此后可使用multimap，定义一个sortmap，此时key为单词出现的次数，value为单词，multimap默认是排升序，我们这里可以多传一个参数greater，要求其按降序排列。
• 3、最后将前k个高频单词取出放入vector中返回即可
  

1.3 代码实现

class Solution {
public:
    vector<string> topKFrequent(vector<string>& words, int k) {
        //1.[] 2.find insert 3.insert
        map<string,int> countmap; 
        //将字符串按照字母顺序排序，并统计出字符串出现的次数                
        for(auto& e : words)
            countmap[e]++;
        
        multimap<int,string,greater<int>> sortmap;//出现次数为key，字符串为value
        //按照出现次数从大到小进行排序
        for(const auto& kv:countmap)
        {
            sortmap.insert(make_pair(kv.second,kv.first));
        }
        
        vector<string> v;
        multimap<int,string>::iterator it = sortmap.begin();
        while(it != sortmap.end() && k--)
        //遍历排好序的map取出前k个出现频率最高的字符串放入vector中
        {
            v.push_back(it->second);
            ++it;
        }        
        return v;  
    }
};

二、前k个高频元素LeetCode347题

2.1 题目描述

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

示例 1:

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

示例 2:

输入: nums = [1], k = 1
输出: [1]

说明：
你可以假设给定的 k 总是合理的，且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。

2.2 解题思路

C++实现，这里使用STL中的map，键值对的思想在这很适用

• 1、首先统计每个数字出现的次数，定义一个countmap，此时key为数字，value数字出现的次数，此时数字已经被排序
• 2、由于出现的次数可能重复，因此后可使用multimap，定义一个sortmap，此时key为数字出现的次数，value为数字，multimap默认是排升序，我们这里可以多传一个参数greater，要求其按降序排列。
• 3、最后将前k个高频元素取出放入vector中返回即可

2.3 代码实现

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int,int> countmap;
        vector<int> v;
        for(auto& e : nums)
        {
            countmap[e]++;
        }
        
        multimap<int,int,greater<int>> sortmap;
        for(auto& kv : countmap)
        {
            sortmap.insert(make_pair(kv.second,kv.first));
        }
        
        auto it = sortmap.begin();
        while(it != sortmap.end() && k--)
        {
            v.push_back(it->second);
            ++it;
        }
        return v;
    }
};