警告
本文最后更新于 2020-03-28,文中内容可能已过时。
Trie又称字典树,是N叉树的一种特殊形式。是一种存储大量字符串的树形数据结构,相当于HashMap
,在存储单词中又比HashMap
节省了很多的内存空间。trie
树的根节点不存储内容,内容都在孩子节点上。
这是一个保存了8个单词的字典树的结构,8个单词分别是"A", “to”, “tea”, “ted”, “ten”, “i”, “in”, “inn”。每条从根节点到叶子节点的路径都构成了单词(有的不需要走到叶子节点也是单词,比如i
, in
),从根节点出发的路径都是给定单词列表的某个/某些单词的前缀,反之如果在某个字符串中没有出现在这棵树的路径上,就说明组不成单词。
通常Trie树也会用在搜索引擎上,就是输入某些词,就会扩展出一些模糊词汇
Golang 实现的写法如下:
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
|
type Trie struct {
children [26]*Trie
isWord bool
}
// Insert
func (t *Trie) Insert(word string) {
cur := t
for k, v := range word {
c := v - 'a'
if cur.children[c] == nil {
cur.children[c] = &Trie{}
}
cur = cur.children[c]
// end
if k == len(word) -1 {
cur.isWord = true
}
}
}
// Search
func (t *Trie) Search(word string) bool {
cur := t
for _, v := range word {
c := v - 'a'
if cur.children[c] == nil {
return false
}
cur = cur.children[c]
}
return cur.isWord
}
// StartWith
func (t *Trie) StartWith(prefix string) bool {
cur := t
for _, v := range prefix {
c := v - 'a'
if cur.children[c] == nil {
return false
}
cur = cur.children[c]
}
return true
}
func main() {
trie := new(Trie)
trie.Insert("inn")
fmt.Println(trie.Search("inn")) // true
fmt.Println(trie.Search("in")) // false
fmt.Println(trie.StartWith("in")) // true
fmt.Println(trie.StartWith("ia")) // false
}
|
关于自定义排序,用这个就可以了
在官方库sort.go
中关于Interface
定义如下:
1
2
3
4
5
6
7
8
9
10
11
12
|
// A type, typically a collection, that satisfies sort.Interface can be
// sorted by the routines in this package. The methods require that the
// elements of the collection be enumerated by an integer index.
type Interface interface {
// Len is the number of elements in the collection.
Len() int
// Less reports whether the element with
// index i should sort before the element with index j.
Less(i, j int) bool
// Swap swaps the elements with indexes i and j.
Swap(i, j int)
}
|
所以我们只需要实现上面三个方法就可以了,如下是一个按照数组中字符串长度排序的一个例子
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
|
package main
import (
"fmt"
"sort"
)
type MySliceSort []string
func (m MySliceSort) Len() int {
return len(m)
}
func (m MySliceSort) Less(i, j int) bool {
return len(m[i]) > len(m[j])
}
func (m MySliceSort) Swap(i, j int) {
m[i], m[j] = m[j], m[i]
}
func main() {
sliceSort := MySliceSort{
"这是五个字",
"我要变六个字",
"三个字",
}
sort.Sort(sliceSort)
fmt.Println(sliceSort) // [我要变六个字 这是五个字 三个字]
}
|