# [leetcode] 208. Implement Trie (Prefix Tree)

188人阅读 评论(0)

Implement a trie with insertsearch, and startsWith methods.

Note:
You may assume that all inputs are consist of lowercase letters a-z.

Trie没有析构函数，这里是需要析构函数的，所以加入了采用深度优先搜索的析构函数。其他部分应该没什么难度了，具体代码：

class TrieNode {
public:
vector<TrieNode*> child;
bool isIn;
// Initialize your data structure here.
TrieNode() : child(vector<TrieNode*>(26, NULL)), isIn(false) {}
};

class Trie {
public:
Trie() : root(new TrieNode()) {}

~Trie() {
delNode(root);
}

void delNode(TrieNode* n) {
for(int i=0; i<26; ++i) {
if(n->child[i]) delNode(n->child[i]);
}
delete n;
}

// Inserts a word into the trie.
void insert(string word) {
TrieNode* n = root;
for(auto ch:word) {
if(n->child[ch-'a'] == NULL)
n->child[ch-'a'] = new TrieNode();
n = n->child[ch-'a'];
}
n->isIn = true;
}

// Returns if the word is in the trie.
bool search(string word) {
TrieNode* n = root;
for(auto ch:word) {
if(n->child[ch-'a'] == NULL) return false;
n = n->child[ch-'a'];
}
return n->isIn;
}

// Returns if there is any word in the trie
// that starts with the given prefix.
bool startsWith(string prefix) {
TrieNode* n = root;
for(auto ch:prefix) {
if(n->child[ch-'a'] == NULL) return false;
n = n->child[ch-'a'];
}
return true;
}

private:
TrieNode* root;
};

// Your Trie object will be instantiated and called as such:
// Trie trie;
// trie.insert("somestring");
// trie.search("key");

0
0

* 以上用户言论只代表其个人观点，不代表CSDN网站的观点或立场
个人资料
• 访问：63283次
• 积分：3078
• 等级：
• 排名：第11733名
• 原创：255篇
• 转载：6篇
• 译文：0篇
• 评论：15条
文章分类
最新评论