LeetCode 49 - Group Anagrams

leetcode_49.go

package main

import "fmt"

func main() {
	l := []string{"eat", "tea", "tan", "ate", "nat", "bat"}
	
	fmt.Printf("%v\n", groupAnagrams(l))
}

func groupAnagrams(l []string) (res [][]string) {
	m := make(map[rune]int)

	for _, a := range l {
		gi := 0

		for _, r := range a {
			i, ok := m[r]
			if !ok {
				i = len(res)
				m[r] = i
			}

			if i > gi {
				gi = i
			}
		}

		if gi >= len(res) {
			res = append(res, []string{a})
		} else {
			res[gi] = append(res[gi], a)
		}
	}

	return res
}