对于以下trie结构。
struct Trie {
bool eow; //when a Trie field isWord = true, hence there is a word
char letter;
Trie *letters[27];
};
我正在尝试为自动完成程序创建一个函数,它基本上在给定特定字符串前缀的trie中打印出单词
这就是我所拥有的:
int wordcheck( TrieNode &node )
{
if (node.isWord == 1) // you have found your word, so return true
{
return 1;
}
for (int i = 0; i < 26; i++)
{
if (node.letters[i] != NULL && wordcheck(*(node.letters[i])))
{
return 1;
}
}
return 0;
}
string find (TrieNode &node, const string &word, string acc)
{
if (word.length() == 0)
{
string x = "";
if (node.isWord == 1){
x = " ";
int check = 1;
for(int i = 0; i < 26; i++)
{
if (node.letters[i] != NULL && wordcheck(*(node.letters[i])))
{
x = x + acc; check = 0; break;
}
}
if(check == 1)
{ return x; }
}
for (int i = 0; i < 26; i++){
if (node.letters[i] != NULL && wordcheck(*(node.letters[i])))
{
char let = (char)(i + (int)'a');
if (x[x.length() - 1 ] == ' ')
{
x = x + acc;
}
x = x + node.letters[i]->letter
+ find(*(node.letters[i]), word, acc + node.letters[i]->letter);
}
}
return x;
}
else if (node.letters[word[0] - 'a'] == NULL)
{ return ""; }
else {
return word[0] + find(*(node.letters[ word[0] - 'a']),
word.substr(1, word.length()-1),
acc + word[0]);
}
}
它似乎工作以外的事实如果我给它一个长前缀它将打印短于前缀的单词。我使用累积递归,我确信有一种更有效的方法来做到这一点。我的问题是,是否有人能够这样做,以便我返回正确的字符串,或者如果可能的话,引导我通过更简单的算法?
答案 0 :(得分:0)
我正在尝试为自动完成程序创建一个函数,它基本上在给定特定字符串前缀的trie中打印出单词
我不打算分析你的程序 - 对我来说这太复杂了,例如我不知道wordcheck
应该做什么?为什么不是bool
而是int
?你真的需要检查你的子系列是否有任何单词,你真的有非空的Trie而没有单词吗?
首先 - 打印所有以给定前缀开头的单词 - 您需要转到所有这些单词开头的节点:
TrieNode* TreeNode::get(std::string word)
{
TreeNode* retVal = this;
for (size_t i = 0; i < word.length(); ++i) {
if (Words[i] < 'a' || words[i] > 'z')
throw std::runtime_error("Wrong word");
if (retVal->letters[word[i] - 'a'] != NULL)
retVal = retVal->letters[word[i] - 'a'];
else
return nullptr;
}
return retVal;
}
您需要打印给定节点中所有单词的函数:
void TreeNode::printAll(std::ostream& os, std::string prefix)
{
if (isWord)
os << prefix << "\n";
for (size_t i = 0; i < 26; ++i) {
if (retVal->letters[i] != NULL)
// this recursive call can be replaced with iterative solution with stack
letters[i]->print(os, prefix + char('a' + i));
}
}
结合这些功能 - 为您提供所需的内容:
void TreeNode::printBeginWith(std::ostream& os, std::string prefix)
{
TreeNode* node = get(prefix);
if (node)
node->printAll(os, prefix);
}