LeetCode-208. 实现 Trie (前缀树)

图论#

Trie（发音类似 “try”）或者说 前缀树 是一种树形数据结构，用于高效地存储和检索字符串数据集中的键。这一数据结构有相当多的应用情景，例如自动补全和拼写检查。

请你实现 Trie 类：

Trie() 初始化前缀树对象。

void insert(String word) 向前缀树中插入字符串 word 。

boolean search(String word) 如果字符串 word 在前缀树中，返回 true（即，在检索之前已经插入）；否则，返回 false 。

boolean startsWith(String prefix) 如果之前已经插入的字符串 word 的前缀之一为 prefix ，返回 true ；否则，返回 false 。

示例：

1
输入
2
["Trie", "insert", "search", "search", "startsWith", "insert", "search"]
3
[[], ["apple"], ["apple"], ["app"], ["app"], ["app"], ["app"]]
4
输出
5
[null, null, true, false, true, null, true]
6

7
解释
8
Trie trie = new Trie();
9
trie.insert("apple");
10
trie.search("apple");   // 返回 True
11
trie.search("app");     // 返回 False
12
trie.startsWith("app"); // 返回 True
13
trie.insert("app");
14
trie.search("app");     // 返回 True

BFS的方法

1
type Trie struct {
2
    children [26]*Trie
3
    isEnd    bool
4
}
5

6
func Constructor() Trie {
7
    return Trie{}
8
}
9

10
func (this *Trie) Insert(word string) {
11
    node := this
12
    for _, ch := range word {
13
        idx := ch - 'a'
14
        if node.children[idx] == nil {
15
            node.children[idx] = &Trie{}
16
        }
17

18
        node = node.children[idx]
19
    }
20

21
    node.isEnd = true
22
}
23

24
func (this *Trie) Search(word string) bool {
25
    node := this
26
    for _, ch := range word {
27
        idx := ch - 'a'
28
        if node.children[idx] == nil {
29
            return false
30
        }
31
        node = node.children[idx]
32
    }
33

34
    return node.isEnd
35
}
36

37
func (this *Trie) StartsWith(prefix string) bool {
38
    node := this
39
    for _, ch := range prefix {
40
        idx := ch - 'a'
41
        if node.children[idx] == nil {
42
            return false
43
        }
44
        node = node.children[idx]
45
    }
46
    return true
47
}