构造字典树

字典树,顾名思义,是关于“字典”的一棵树。即:它是对于字典的一种存储方式(所以是一种数据结构而不是算法)。这个词典中的每个“单词”就是从根节点出发一直到某一个目标节点的路径,路径中每条边的字母连起来就是一个单词。

构造字典树_第1张图片
(标橙色的节点是“目标节点“,即根节点到这个目标节点的路径上的所有字母构成了一个单词。)

每个小⚪都是一个节点node(参考链表),node中包含指向的下一个节点Nextnode,以及是否是单词的结尾flag

class Trie{
private:
bool flag;//是否为尾节点
Trie* next[R];//表示的是指向R个子节点的链接。比如我们现在要表示英语单词的话,R就为26。
}

构造字典树_第2张图片

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;
    }
};

你可能感兴趣的:(数据结构,开发语言,数据结构)