692. Top-K Frequent Words

692-top-K-frequent-words.py

from collections import Counter
import heapq
from typing import List


class Solution:
    def topKFrequent(self, words: List[str], k: int) -> List[str]:
        """

        https://leetcode.com/problems/top-k-frequent-words
        Args:
            words: List[str]
            k: int

        Returns:
            List[str] of the k most frequent word in words

        """
        count = Counter(words)  # O(n) runtime, O(n) space
        heap = (heapq.nsmallest(k, count.items(), key=lambda item: (-item[1], item[0])))  # O(n*log[k]) runtime, O(n) space
        return [word for word, _ in heap]  # O(k) runtime, O(k) space