How would you go about writing a function that can find words in a list that are anagrams of another word in that same list?

Question

Anonymous · Accepted Answer

from collections import defaultdict

def find_anagrams(words):
    # Dictionary to group words by their sorted character tuple
    anagram_dict = defaultdict(list)
    
    # Group words by their sorted character sequence
    for word in words:
        sorted_word = ''.join(sorted(word))
        anagram_dict[sorted_word].append(word)
    
    # Collect words that have anagrams (more than 1 entry in the group)
    result = []
    for group in anagram_dict.values():
        if len(group) > 1:  # Only keep groups that have more than 1 word
            result.extend(group)
    
    return result

# Example usage
words = ["listen", "silent", "enlist", "google", "gigolo", "dog", "god"]
anagrams = find_anagrams(words)
print(anagrams)

How would you go about writing a function that can find words in a list that are anagrams of another word in that same list?

Practice this question with AI

Go Premium

Practice More Questions

Community Answers

Unlock Community Insights

Everything You Need to Ace DS Interviews