我有一个Trie数据结构,它眨眼间搜索了10万个元素。然而,它只搜索以搜索到的字符串开头的单词,例如“Fi”将找到“Final”但不会找到“GooFi”,我希望它也返回“GooFi”。这就是为什么我在这里问你们,在这种情况下这是否是正确的结构。我自己实现了它,写了单元测试,所以它到目前为止工作。我需要的是暗示我的目标是如何实现的,我不希望任何人为我编写代码,这不是我在这里的原因。无论如何这是我的搜索实现:
public List<string> Seach(string word)
{
List<string> results = new List<string>();
this.DoSearch(this.Root, 0, new StringBuilder(word), results);
return results;
}
private void DoSearch(TrieNode currentNode, int currentPositionInWord, StringBuilder word, List<string> results)
{
if (currentPositionInWord >= word.Length)
{
this.DfsForAllWords(currentNode, word, results);
return;
}
char currentChar = word[currentPositionInWord];
bool containsKey = currentNode.Children.ContainsKey(currentChar);
if (containsKey)
{
if (currentPositionInWord == word.Length - 1)
{
results.Add(word.ToString());
}
TrieNode child = currentNode.Children[currentChar];
this.DoSearch(child, ++currentPositionInWord, word, results);
}
}
private void DfsForAllWords(TrieNode currentNode, StringBuilder word, List<string> results)
{
foreach (var node in currentNode.Children.ToArray())
{
word.Append(node.Value.Value);
if (node.Value.IsWord)
{
results.Add(word.ToString());
}
this.DfsForAllWords(node.Value, word, results);
word.Length--;
}
}
非常感谢任何帮助。
答案 0 :(得分:1)
您可以在所有节点上使用某种索引。
Dictionary<char,List<TrieNode>> nodeIndex;
现在,如果您想搜索“Fi”的示例,请迭代nodeIndex
并像之前一样进行搜索。如果您在该迭代中找到了某些内容,则必须在找到的子字符串前面加上指向实际节点的字符串。
public List<string> Seach(string word)
{
List<string> results = new List<string>();
foreach(var node in nodeIndex[word[0]])
{
List<string> nodeResults = new List<string>();
this.DoSearch(node, 0, new StringBuilder(word), nodeResults);
foreach(var nodeResult in nodeResults)
{
var text = string.Format("{0}{1}",node.Parent.Text, nodeResult);
results.Add(node.Parent.Text, nodeResult);
}
}
return results.Distinct().ToList();
}
也许你必须要实现一些缺失的属性。
答案 1 :(得分:0)
https://github.com/gngeorgiev/Trie
如果有人需要,这是Trie的回购。支持前缀和子字符串搜索。