leetcode-692. 前K个高频单词

给一非空的单词列表，返回前 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 次。

思路：map存储key与value，对于返回的List集合，排序方式由两个优先级判断，分别是出现次数和字典序，我们可以通过构建大小为k的小顶堆返回（重写比较方法），或直接重写map集合里的sort方法

法一：（时间复杂度：logk*n）

public class Solution {
     public List<String> topKFrequent(String[] words, int k) {
       
     Map<String, Integer> map = new HashMap();
        for (String word : words) {
            map.put(word, map.getOrDefault(word, 0) + 1);
        }
        PriorityQueue<String> queue =new PriorityQueue<String>(
         (a,b)->{
             if(map.get(a).equals(map.get(b))){//==与equals
                return b.compareTo(a);
             }else{
                return map.get(a)-map.get(b);//建立小根堆
             }
         });
         for(Map.Entry<String,Integer>entry:map.entrySet()){
             queue.offer(entry.getKey());
             if(queue.size()>k){
                 queue.poll();//弹出最小的元素
             }
         }
         LinkedList<String> list=new LinkedList<String>();
         while(queue.size()>0){//将小根堆里元素弹出即可
             list.add(queue.poll());
         }

        Collections.reverse(list);
         return list;

     } 
}

注意：

1. 我们在添加元素时，可以通过getOrDefault（key，defaultvalue）来简化if-else的书写
2. 注意书写Lambda表达式体

     (a,b)->{
             if(map.get(a).equals(map.get(b))){//出现频率相等时
                return b.compareTo(a);//以字典序降序排列，注意之后要反转
             }else{
                return map.get(a)-map.get(b);//建立小根堆
             }

法2：map中重写接口
注意，我们这里主要返回的是map中的key

public class Solution {
     public List<String> topKFrequent(String[] words, int k) {
         HashMap <String,Integer>map =new HashMap<>();
         for(String str:words){//添加至map
             map.put(str,map.getOrDefault(str,0)+1);
         }
         LinkedList<String> list=new LinkedList<String>();//存储返回的Key
         for(String str2:map.keySet()){
             list.add(str2);
         }
         list.sort((a,b)->{
             if(map.get(a).equals(map.get(b))){
                 //值相同时，返回字典序小的
                 return a.compareTo(b);
             }else{
                 return map.get(b)-map.get(a);//降序排序出现的次数
             }
         });

         return list.subList(0,k);
     } 
}