107.(208)实现 Trie (前缀树)

    技术2025-04-12  15

    题目描述:

    实现一个 Trie (前缀树),包含 insert, search, 和 startsWith 这三个操作。

    示例:

    Trie trie = new Trie();

    trie.insert("apple"); trie.search("apple");   // 返回 true trie.search("app");     // 返回 false trie.startsWith("app"); // 返回 true trie.insert("app");    trie.search("app");     // 返回 true说明:

    你可以假设所有的输入都是由小写字母 a-z 构成的。 保证所有输入均为非空字符串。

    代码:

    class Trie { Trie* next[26]; bool end; public: /** Initialize your data structure here. */ Trie() { for(int i=0;i<26;i++) { next[i]=NULL; end=false; } } /** Inserts a word into the trie. */ void insert(string word) { Trie *root=this; int loc; for(int i=0;i<word.length();i++) { loc=word[i]-'a'; if(root->next[loc]!=NULL) root=root->next[loc]; else { Trie* temp=new Trie(); root->next[loc]=temp; root=temp; } } root->end=true; } /** Returns if the word is in the trie. */ bool search(string word) { Trie *root=this; int loc; for(int i=0;i<word.length();i++) { loc=word[i]-'a'; if(root->next[loc]==NULL) return false; else root=root->next[loc]; } return root->end; } /** Returns if there is any word in the trie that starts with the given prefix. */ bool startsWith(string prefix) { Trie *root=this; for(int i=0;i<prefix.length();i++) { int loc=prefix[i]-'a'; if(root->next[loc]==NULL) return false; else root=root->next[loc]; } 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); */

    执行效率:

    执行用时:108 ms, 在所有 C++ 提交中击败了83.20%的用户

    内存消耗:43.4 MB, 在所有 C++ 提交中击败了54.55%的用户

    Processed: 0.011, SQL: 9