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