-
Notifications
You must be signed in to change notification settings - Fork 215
/
Copy pathgroup_anagrams.go
65 lines (49 loc) · 2.73 KB
/
group_anagrams.go
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
59
60
61
62
63
64
65
/*
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
*/
package main
import "sort"
// `GroupAnagrams` groups anagrams together.
func GroupAnagrams(words []string) [][]string {
// Create a map where the keys are sorted strings of words and the values are lists of words
// that have the same sorted string.
anagrams := map[string][]string{}
for _, word := range 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 map.
anagrams[sortedWord] = append(anagrams[sortedWord], word)
}
// Create a list of lists, where each inner list contains all the anagrams of a word in the original list.
result := [][]string{}
for _, group := range anagrams {
result = append(result, group)
}
// Return the list of lists.
return result
}
// `sortWord` takes a word as input and returns a string that contains the word's letters in sorted order.
func sortWord(word string) string {
// Convert the word to a byte array.
wordBytes := []byte(word)
// Sort the byte array.
sort.Slice(wordBytes, func(i, j int) bool {
return wordBytes[i] < wordBytes[j]
})
// Return a string that contains the sorted byte array.
return string(wordBytes)
}