实现前缀树
Trie (发音为 “try”) 或前缀树是一种树数据结构,用于检索字符串数据集中的键。这一高效的数据结构有多种应用:
- 自动补全
- 拼写检查
- 最长前缀匹配
- 打字预测
代码
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个节点。