41 lines
650 B
Go
41 lines
650 B
Go
|
package anagram
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bufio"
|
||
|
|
"io"
|
||
|
|
"sort"
|
||
|
|
|
||
|
|
"github.com/ray1729/puzzle-solver/util"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DB interface {
|
||
|
|
FindAnagrams(s string) []string
|
||
|
|
}
|
||
|
|
|
||
|
|
func toKey(s string) string {
|
||
|
|
xs := util.LowerCaseAlpha(s)
|
||
|
|
sort.Slice(xs, func(i, j int) bool { return xs[i] < xs[j] })
|
||
|
|
return string(xs)
|
||
|
|
}
|
||
|
|
|
||
|
|
type HashDBImpl map[string][]string
|
||
|
|
|
||
|
|
func Load(r io.Reader) (DB, error) {
|
||
|
|
db := make(HashDBImpl)
|
||
|
|
sc := bufio.NewScanner(r)
|
||
|
|
for sc.Scan() {
|
||
|
|
s := sc.Text()
|
||
|
|
k := toKey(s)
|
||
|
|
db[k] = append(db[k], s)
|
||
|
|
|
||
|
|
}
|
||
|
|
if err := sc.Err(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return db, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d HashDBImpl) FindAnagrams(s string) []string {
|
||
|
|
return d[toKey(s)]
|
||
|
|
}
|