前 k 个高频单词

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

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

示例 1：

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

链接：https://leetcode-cn.com/problems/top-k-frequent-words

求解思路：

1. 用 Map 将每个单词以及出现的次数统计；
2. 将这些单词放入一个 ArrayList 中，然后进行排序；
3. 在排序方法 sort() 的第二个参数中需传入一个 Comparator 对象，用来指定排序的比较规则（如果两单词出现的次数相等，就按照字符串的字典序来排序，即可使用CompareTo()方法）；
4. 最后使用方法 subList（0，k）取前 k 个高频单词。

代码实现：

class Solution {
   public List<String> topKFrequent(String[] words, int k) {
        //1.用Map统计words中每个单词出现的次数
        Map<String,Integer> map = new HashMap<>();
        for(String word : words){
            Integer value = map.get(word);
            if(value == null){
                map.put(word,1);
            }else{
                map.put(word,value + 1);
            }
        }
        //2.将所有的key取出来放到一个List中
        List<String> wordList = new ArrayList<>(map.keySet());
        //3.对 wordList 进行排序。
        // 需在 sort 方法的第二个参数中传入一个Comparator对象，用来指定比较规则
        Collections.sort(wordList, new Comparator<String>() { //java的匿名内部类
            @Override
            public int compare(String o1, String o2) {
                Integer count1 = map.get(o1);
                Integer count2 = map.get(o2);
                if(count1.equals(count2)){//如果出现次数相等，就按字符串的字典序来排序
                    return o1.compareTo(o2);
                }
                return count2 - count1;
            }
        });
        return wordList.subList(0,k);
    }
}