leetcode 692. Top K Frequent Words（频率最高的K个单词）

最新推荐文章于 2025-07-13 21:05:47 发布

蓝羽飞鸟

最新推荐文章于 2025-07-13 21:05:47 发布

阅读量372

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode 算法

本文链接：https://blog.youkuaiyun.com/level_code/article/details/127401515

leetcode 专栏收录该内容

490 篇文章

订阅专栏

该博客介绍了一种统计字符串数组中单词频率的方法，通过优先队列和哈希映射来找出出现最频繁的K个单词。首先，利用哈希映射统计每个单词的出现次数，然后用优先队列按频率降序及字典序升序对单词进行排序，最终取出前K个高频单词。此方法适用于数据结构和算法的学习与实践。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an array of strings words and an integer k, return the k most frequent strings.

Return the answer sorted by the frequency from highest to lowest. Sort the words with the same frequency by their lexicographical order.

Example 1:

Input: words = [“i”,“love”,“leetcode”,“i”,“love”,“coding”], k = 2
Output: [“i”,“love”]
Explanation: “i” and “love” are the two most frequent words.
Note that “i” comes before “love” due to a lower alphabetical order.
Example 2:

Input: words = [“the”,“day”,“is”,“sunny”,“the”,“the”,“the”,“sunny”,“is”,“is”], k = 4
Output: [“the”,“is”,“sunny”,“day”]
Explanation: “the”, “is”, “sunny” and “day” are the four most frequent words, with the number of occurrence being 4, 3, 2 and 1 respectively.

统计words数组中各单词出现的频率，返回频率由高到低的前K个单词。
频率一样时，按字典顺序排序。

思路：

首先要统计每个单词出现的频率，所以要用到hash map.
遍历一遍words，保存每个单词和对应的频率到hash map.

然后要按频率由高到低排序，找到对应的单词。
这步可以用优先队列，定义排序规则：按频率由高到低，频率相同时，按字典顺序排序。

最后只需要从优先队列中取出前k个单词装入list。

可以自定义一个class，包含单词和对应的频率
字典顺序的比较可以用String的compareTo方法。

public List<String> topKFrequent(String[] words, int k) {
    Queue<Node> heap = new PriorityQueue<>(k, (a, b)->(a.freq == b.freq ? 
                                                      a.word.compareTo(b.word) : b.freq - a.freq));
    HashMap<String, Integer> map = new HashMap<>();
    List<String> res = new ArrayList<>();
    
    for(String word : words) {
        map.put(word, map.getOrDefault(word, 0)+1);
    }
    
    map.forEach((key, val) -> {
        heap.offer(new Node(key, val));
    });
    
    while(k > 0) {
        res.add(heap.poll().word);
        k --;
    }
    return res;
}

class Node{
    String word;
    int freq;
    
    public Node(String word, int freq) {
        this.word = word;
        this.freq = freq;
    }
}

不过自定义的class没有map中的entrySet高效

public List<String> topKFrequent(String[] words, int k) {
    Queue<Map.Entry<String,Integer>> heap = new PriorityQueue<>((a, b)->(a.getValue() == b.getValue() ? 
                                                      a.getKey().compareTo(b.getKey()) : b.getValue() - a.getValue()));
    HashMap<String, Integer> map = new HashMap<>();
    List<String> res = new ArrayList<>();
    final int num = k;
    
    for(String word : words) {
        map.put(word, map.getOrDefault(word, 0)+1);
    }
    
    for(Map.Entry<String, Integer> entry : map.entrySet()) {
        heap.offer(entry);
    }
    
    while(k > 0) {
        res.add(heap.poll().getKey());
        k --;
    }
    return res;
}