字典树,顾名思义,是关于“字典”的一棵树。即:它是对于字典的一种存储方式(所以是一种数据结构而不是算法)。这个词典中的每个“单词”就是从根节点出发一直到某一个目标节点的路径,路径中每条边的字母连起来就是一个单词。
(标橙色的节点是“目标节点“,即根节点到这个目标节点的路径上的所有字母构成了一个单词。)
每个小⚪都是一个节点node(参考链表),node中包含指向的下一个节点Nextnode,以及是否是单词的结尾flag
class Trie{
private:
bool flag;//是否为尾节点
Trie* next[R];//表示的是指向R个子节点的链接。比如我们现在要表示英语单词的话,R就为26。
}
class Trie{
private:
// 定义节点
bool flag = false;
Trie* next[26] = {nullptr};//指向26个空节点
public:
Trie(){} // 构造函数
void insert(string word){ // 插入
Trie* node = this;
for(int i=0; i<word.length(); i++){
char c = word[i];
if(node->next[c-'a']==nullptr){
node->next[c-'a'] = new Trie();
}//对这个字母继续拓展新Trie
node = node->next[c-'a'];
}
node->flag = true; // 单词串的尾节点为true
}
bool search(string word){ // 查找单词是否存在Trie中
Trie* node = this;
for(int i=0; i<word.length(); i++){
char c = word[i];
if(node->next[c-'a']==nullptr) return false;//根节点是空指针
node = node->next[c-'a'];
}
return node->flag; // 如果该节点是串尾节点,则为true
}
bool starsWith(string prefix){ //查找前缀
Trie * node = this;
for(int i=0; i<prefix.length(); i++){
char c = prefix[i];
if(node->next[c-'a']==nullptr) return false;
node = node->next[c-'a'];
}
return true;
}
};