有效搜索包含

时间:2013-08-05 17:11:26

标签: c# string search trie

我有一个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--;
    }
}

非常感谢任何帮助。

2 个答案:

答案 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的回购。支持前缀和子字符串搜索。