LeetCode 208. 实现 Trie (前缀树)

LeetCode 208. 实现 Trie (前缀树)
此题用到了前缀和的知识点 可以把每一个字母都模拟成一个节点,那么一个节点最多有26个子节点
(26个字母) 可以创建一个结构体 里面包含着26个字母。
插入操作 :
把一个单词插入到前缀树里面去 如果就节点就往下遍历 如果没有节点就创建节点并遍历到创建节点
搜索操作:
如果有节点 就继续往下遍历 而且遍历到的最后一个节点必须有结束标志
startsWith:
和搜索操作相同 遍历到最后不需要判断是否存在结束标志

class Trie {
public:
    struct Node{
        bool is_end;
        Node *node[26];
        Node()
        {
            is_end =false;
            for(int i=0;i <26 ;i++)
            {
                node[i] =NULL;
            }
        }
    };
    Node * root;
    Trie() {
        root = new Node();
    }
    
    void insert(string word) {
        if(!root ) return ;
        auto p=root;
        for(auto u :word)
        {
            int  c = u -'a';
            if( !p->node[c])p->node[c] =new Node();
            p= p->node[c];
        }
        p->is_end =true;
    }
    
    bool search(string word) {
        if(!root) return false;
        auto p =root;
        for(auto u :word)
        {
            int c= u-'a';
            if(!p->node[c]) return false;
            p = p ->node[c];
        }
        return p ->is_end;
    }

    
    bool startsWith(string prefix) {
        if(!root) return false;
        auto p =root;
        for(auto u :prefix)
        {
            int c= u-'a';
            if(!p->node[c]) return false;
            p = p ->node[c];
        }
        return true;
    }
};

/**
 * 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);
 */

你可能感兴趣的:(每日一题,leetcode,算法,前缀树)