void Trie::AddWord(string s) {
TrieNode* current = root;
if(s.length() == 0)
{
current->Mark(); // recognize empty word
return;
}
unsigned long len = s.length();
for (int i=0; i<len-1; i++) {
TrieNode *child = current->FindChild(s[i]);
if(child != NULL)
{
current = child;
}
else {
TrieNode* tmp = new TrieNode(s[i], false);
tmp->Set(s[i]);
current->AddChild(tmp);
current = tmp;
}
if (s[i+1]=='/') {
current->Mark();
break;
}
if ( i == len - 2 )
current->Mark();
}
}
// Inserts a word into the trie.
void insert(string s)
{
TrieNode *prev = nullptr;
TrieNode *curr = m_root;
// We check the characters of s one by one. Note that
// we also add the NULL terminator to the Trie which
// indicates the end of a word.
for (int i = 0; i <= s.length(); i++)
{
prev = curr;
curr = prev->FindChild(s[i]);
// Can't find a TrieNode with the character s[i],
// so create a new one and add it to the children
// list of prev.
if (curr == nullptr)
{
curr = new TrieNode(s[i]);
prev->AddChild(curr);
}
}
}
