实现前缀树

实现前缀树

Trie (发音为 “try”) 或前缀树是一种树数据结构，用于检索字符串数据集中的键。这一高效的数据结构有多种应用：

1. 自动补全
2. 拼写检查
3. 最长前缀匹配
4. 打字预测

代码

private:
bool isEnd = false;
Trie *next[26] = {nullptr};
public:
/** Initialize your data structure here. */
Trie() {
    
}

/** Inserts a word into the trie. */
void insert(string word) {
    Trie *node = this;
    for(char c:word)
    {
        if(node->next[c-'a']==NULL)  //如果没有下一个字母节点
        node->next[c-'a'] = new Trie();
        node = node->next[c-'a'];
    }
    node->isEnd = true;
}

/** Returns if the word is in the trie. */
bool search(string word) {
    Trie *node = this;
    for(char c:word)
    {
        node = node->next[c-'a'];
        if(node == NULL) return false;
    }
    return node->isEnd;
}

/** Returns if there is any word in the trie that starts with the given prefix. */
bool startsWith(string prefix) {
    Trie *node = this;
    for(char c:prefix)
    {
        node = node->next[c-'a'];
        if(node == NULL) return false;
    }
    return true;
}

总结

• 时间复杂度O(M),M为键长。
• 空间复杂度O(M），最坏的情况，已插入的键与新键没有公共前缀，此时需要添加M个节点。