leetcode 692. 前K个高频单词

1. 题目链接 https://leetcode-cn.com/problems/top-k-frequent-words/

2. 题目描述

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

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

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

3. 解题思路

   1. 基于词频排序，再去前k个元素
   2. 基于词频构建小顶堆，弹出k个元素

4. 代码

   1. python

      class Solution:
          def topKFrequent(self, words, k):
              # from collections import Counter
              # c = Counter(words)
              # return sorted(list(c.keys()), key=lambda a: (-c[a], a))[:k]
              from collections import Counter
              import heapq
              heap = [(-j, i) for i, j in Counter(words).items()]
              heapq.heapify(heap)
              return [heapq.heappop(heap)[1] for _ in range(k)]