K Most Frequent Words
You have been given an array/list 'WORDS' of 'N' non-empty words, and an integer 'K'. Your task is to return the 'K' most frequent words sorted by their frequency from highest to lowest.

Note:

If two words have the same frequency then the lexicographically smallest word should come first in your answer.

Follow up:

Can you solve it in O(N * logK) time and O(N) extra space?

Input Format:

The first line of input contains two integers, N and K, where N is the number of the words and, K denotes the number of words to return. The next line contains N single space-separated words. Each word consists of only lowercase Latin letters.

Output Format:

For each input, print K single space-separated words, where the ith word denotes the ith most frequent word.

Note:

You are not required to print the expected output; it has already been taken care of. Just implement the function.

Constraints:

1

Question

K Most Frequent Words

You have been given an array/list 'WORDS' of 'N' non-empty words, and an integer 'K'. Your task is to return the 'K' most frequent words sorted by their frequency from highest to lowest.

Note:

If two words have the same frequency then the lexicographically smallest word should come first in your answer.

Follow up:

Can you solve it in O(N * logK) time and O(N) extra space?

Input Format:

The first line of input contains two integers, N and K, where N is the number of the words and, K denotes the number of words to return.

The next line contains N single space-separated words. Each word consists of only lowercase Latin letters.

Output Format:

For each input, print K single space-separated words, where the ith word denotes the ith most frequent word.

Note:

You are not required to print the expected output; it has already been taken care of. Just implement the function.

Constraints:

1 <= N <= 10^5
1 <= K <= number of unique words

Time Limit: 1sec

Accepted Answer

Given an array of words and an integer k, return the k most frequent words sorted by frequency.

Use a hashmap to count the frequency of each word
Use a priority queue to keep track of the k most frequent words
Sort the priority queue based on frequency and lexicographical order

Accepted Answer

Brute force was to iterate on every collection id/name over every file given and then group them up. That was not an optimized solution.

Optimal way is to use Set to remove duplicates first and then use min heap with custom comparators to sort files by sizes, custom class object to group the files by collections.

Accepted Answer

Using Sorting

We use a hashmap to store the frequencies of all unique words in the list, where the key is the word and value is the frequency of that word in the list. While iterating over each word, we will increment the frequency corresponding to that word in the hashmap by 1.

Now, we need to find the K largest frequencies among them. For this, we store each word and its frequency as a pair in a list. We then sort the list based on the criteria that the word having a higher frequency should come first. In case if two words have the same frequency, we compare the words and the lexicographically smaller word should come first.

After sorting, the first K words of the list would be our required answer.

Space Complexity: O(n)Explanation:

O(N), where N is the number of the words.

In the worst case, we will be storing all the N-words in the map, and in the list, both requires O(N) space.

Time Complexity: O(nlogn)Explanation:

O(N * logN), where N is the number of the words.

Hashing each word would require O(N) time. Sorting the list would require O(N * logN) time.

Accepted Answer

Using Min Heap

We use a hashmap to store the frequencies of all unique words in the list, where the key is the word and value is the frequency of that word in the list. While iterating over each word, we will increment the frequency corresponding to that word in the hashmap by 1.

Now, we need to find the K largest frequencies among them. For this, we will first initialize a min-heap of size K that will keep the least frequency word as the top(root) element. When two words have the same frequencies we will keep the lexicographically larger word at the top of other words in the min-heap(as the lexicographically larger word will have the least priority from all top most K words).

Now, iterating through the Hashmap, we will add first K words into the min-heap. For the next words, we will compare the frequency of the current word with the minimum frequency from the heap(top element of min heap). If the current frequency is larger than the minimum frequency from the heap. We will pop the top element from the min-heap and insert our current word into the list. Otherwise, if the current frequency is equal to the minimum frequency then we will compare both the words and only keep the lexicographically smaller word from them.

Now, our min heap will be storing the top K most frequent words. We will pop all words in a list. This list will now contain the least frequent word out of these K words at the 0th position and the most frequent word at the (K - 1)th position. This means we need to return the reverse of this list as our answer.

Space Complexity: O(n)Explanation:

O(N), where N is the total number of words.

In the worst case, we will be storing all the N words in a map that requires O(N) space. Also, O(K) space will be required to maintain a heap of size K.

Time Complexity: OtherExplanation:

O(N * log(K)), where N is the total number of words, and K is the number of words to return.

In the worst case, we will be iterating on all N words in the list and adding it to the min-heap of size K that requires O(logK) time, and when the min-heap is full of its capacity removing one word will also take O(logK) time.

You have been given an array/list 'WORDS' of 'N' non-empty words, and an integer 'K'. Your task is to return the 'K' most frequent words sorted by their frequency from highest to lowest.

Note:

Follow up:

Input Format:

Output Format:

Note:

Constraints:

Top Atlassian SDE-2 interview questions & answers

Popular interview questions of SDE-2