Leetcode208题：实现一个前缀树（Trie）

题目描述

实现一个 Trie (前缀树)，包含 insert, search, 和 startsWith 这三个操作。
示例:
Trie trie = new Trie();
trie.insert(“apple”);
trie.search(“apple”); // 返回 true
trie.search(“app”); // 返回 false
trie.startsWith(“app”); // 返回 true
trie.insert(“app”);
trie.search(“app”); // 返回 true
说明:
你可以假设所有的输入都是由小写字母 a-z 构成的。
保证所有输入均为非空字符串。

我的思路

刚看到这个题，我并不知道什么是前缀树，上网搜了下博客，这篇讲的不错，大体了解了前缀树的概念，并不是很难。
https://blog.csdn.net/weixin_39778570/article/details/81990417
前缀树如下图所示：

题目中需要将word中的每个字符串存入前缀树中，所以先分析前缀树中的每个节点需要包含哪些信息。

 1  需要标识该节点是否为末尾节点
 2. 该节点表示的字符是什么
 3. 该节点的子节点

所以可以用一个bool类型变量标识该节点是否为末尾节点
用一个指针数组，记录该节点的子节点（数组中的每个元素为指针），这里有一个技巧，我们定义一个长度为26的数组：
Trie * next[26];
之所以这样定义，是因为：
1. 每一个节点的子节点都由26种可能性，因为下一个字符可能为a~z中的任意一个字符；
2. 数组的小标从0~26，我们用 ‘s’ = 'a’来计算数组的小标，即数组小标为 0 ~ 26，代表字母 a ~ z;

C++代码如下（注释详细）

```cpp
class Trie {
public:
    /** Initialize your data structure here. */
    bool is_endstr;//用于标记该节点是否为末尾节点
    Trie * next[26];//一个指针数组，存放该节点指向的子节点，长度为26（因为每个节点的子节点最多有26个）
                    //数组的下标为1~26，即代表字母a~z
    Trie() {
        is_endstr = false;
        memset(next, 0, sizeof(next));//每次新建一个Trie节点，都需要将其next数组置为0
    }
    
    /** Inserts a word into the trie. */
    void insert(string word) {
        Trie* cur = this;//根节点
        for (auto w : word)
        {
            if (cur->next[w - 'a'] == NULL)//若字符串word中的字符w对应的节点不存在，则新建一个节点
            {
                Trie* newnode = new Trie();//新建一个节点
                cur->next[w - 'a'] = newnode;//将新节点链接在后面
            }
            cur = cur->next[w - 'a'];//更新当前节点
        }
        cur->is_endstr = true;//遍历完一个字符串之后，就将最后一个字符的is_str设为true，表明到达一个单词的最后一个字符
        
    }
    
    /** Returns if the word is in the trie. */
    bool search(string word) {
        Trie* cur = this;//获取根节点
        for (auto w : word)
        {
            if (cur != NULL)
            {
                cur = cur->next[w - 'a'];
            }
        }
        return (cur != NULL && cur->is_endstr);判断一个word是否在树种，需要满足两个条件，非null和is_str为true
        
    }
    
    /** Returns if there is any word in the trie that starts with the given prefix. */
    bool startsWith(string prefix) {
        Trie* cur = this;
        for (auto w : prefix)
        {
            if (cur != NULL)
            {
                cur = cur->next[w - 'a'];
            }
        }
        return (cur != NULL);//不需要遍历到末尾节点，只要非空即可
    }
};

/**
 * Your Trie object will be instantiated and called as such:
 * Trie* obj = new Trie();
 * obj->insert(word);
 * bool param_2 = obj->search(word);
 * bool param_3 = obj->startsWith(prefix);
 */