-
Notifications
You must be signed in to change notification settings - Fork 215
/
Copy pathgroup_anagrams.py
58 lines (42 loc) · 2.54 KB
/
group_anagrams.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
'''
Group Anagrams
Sample Input: = ["yo", "act", "flop", "tac", "foo", "cat", "oy", "olfp"]
Output: [["yo", "oy"], ["flop", "olfp"], ["act", "tac", "cat"], ["foo"]]
Explanation:
The code snippet is a function that groups anagrams together. An anagram is a word or phrase formed by rearranging
the letters of another word or phrase.
The function first defines two functions: GroupAnagrams and sortWord. The GroupAnagrams function takes a list of words
as input and returns a list of lists, where each inner list contains all the anagrams of a word in the original list.
The sortWord function takes a word as input and returns a string that contains the word's letters in sorted order.
The GroupAnagrams function works by first creating a map where the keys are sorted strings of words and the values are
lists of words that have the same sorted string. Then, the function iterates through the list of words, calling the
sortWord function to get the sorted string for each word. The function then adds the word to the list of words associated with the sorted string in the map. Finally, the function iterates through the map, adding each list of words to a list of lists.
The sortWord function works by converting the word to a byte array and then calling the sort.Slice function to sort
the byte array. The function then returns a string that contains the sorted byte array.
O(w * n * log(n)) time | O(wn) space - where w is the number of words and n is the length of the longest word
'''
from typing import List
from collections import defaultdict
def sortWord(word: str) -> str:
# Sort the characters in the word and return the sorted string
return ''.join(sorted(word))
def GroupAnagrams(words: List[str]) -> List[List[str]]:
# Create a defaultdict where the keys are sorted strings of words and the values are lists of words
# that have the same sorted string.
anagrams = defaultdict(list)
# Iterate through the words
for word in words:
# Get the sorted string for the word
sortedWord = sortWord(word)
# Add the word to the list of words associated with the sorted string in the defaultdict
anagrams[sortedWord].append(word)
# Create a list of lists, where each inner list contains all the anagrams of a word in the original list
result = list(anagrams.values())
# Return the list of lists
return result
# Example usage
words = ["eat", "tea", "tan", "ate", "nat", "bat"]
groupedAnagrams = GroupAnagrams(words)
# Print the grouped anagrams
for group in groupedAnagrams:
print(group)