1

对于以下结构的一个 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]);
 }
}

如果我给它一个长前缀，它似乎可以工作，它会打印比前缀短的单词。我使用了累积递归，我确信有一种更有效的方法可以做到这一点。我的问题是，是否有人可以让我返回正确的字符串，或者如果可能的话指导我通过更简单的算法？

4

1 回答 1

0

我正在尝试为自动完成程序创建一个函数，它基本上在给定特定字符串前缀的情况下打印出 trie 中的单词

我不打算分析你的程序——对我来说太复杂了，例如我不知道wordcheck应该做什么？为什么不是bool但是int？你真的需要检查你的 sub-trie 是否有任何单词，你真的有没有单词的非空 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);
}
于 2012-10-20T11:24:40.670 回答